# Example usage
'''from scripts.run_assessment_prediction_trainer import CompanyModelPipeline
company_ids = ['testid']
input_base_path = '/root/ds_erp_ai/data/raw/erp_assessment_prediction'  # The base path where the raw data for each company is stored

pipeline = CompanyModelPipeline(company_ids=company_ids, input_base_path=input_base_path)
pipeline.run_pipeline()'''

from src.pipeline.inference import AssessmentInference



inference = AssessmentInference(
    company_id="testid",num_assessments=2
)

result = inference.run()


print(result)