diff --git a/modules/ml-pipeline/src/pipeline/4_generate_metrics.py b/modules/ml-pipeline/src/pipeline/4_generate_metrics.py index b9f279d..709ce53 100644 --- a/modules/ml-pipeline/src/pipeline/4_generate_metrics.py +++ b/modules/ml-pipeline/src/pipeline/4_generate_metrics.py @@ -42,7 +42,7 @@ generate_metrics_params = yaml.safe_load(open(generate_metrics_path)) feature_process_path = Path(__file__).parent / "configs" / "feature_processor.yaml" feature_process_params = yaml.safe_load(open(feature_process_path)) -target = (feature_process_params["feature_processor_config"]["target"],) +target = feature_process_params["feature_processor_config"]["target"] test_data_filepath = generate_predictions_params["test_data_filepath"] predictions_output_filepath = generate_predictions_params["predictions_output_filepath"] predictions_column_name = generate_predictions_params["predictions_column_name"]