@@ -914,7 +914,7 @@ def test_prophet_floor_cap(operator_setup, model):
914
914
yaml_i ["spec" ]["model" ] = model
915
915
yaml_i ["spec" ]["historical_data" ] = {"format" : "pandas" }
916
916
yaml_i ["spec" ]["datetime_column" ]["name" ] = HISTORICAL_DATETIME_COL .name
917
- # yaml_i["spec"]["output_directory"]["url"] = operator_setup
917
+ yaml_i ["spec" ]["output_directory" ]["url" ] = operator_setup
918
918
yaml_i ["spec" ]["target_column" ] = "target"
919
919
yaml_i ["spec" ]["model_kwargs" ] = {"min" : 0 , "max" : 20 }
920
920
@@ -969,6 +969,10 @@ def test_generate_files(operator_setup, model):
969
969
operator_config = ForecastOperatorConfig .from_dict (yaml_i )
970
970
results = operate (operator_config )
971
971
files = os .listdir (operator_setup )
972
+ if "errors.json" in files :
973
+ with open (os .path .join (operator_setup , "errors.json" )) as f :
974
+ print (f"Errors in build! { f .read ()} " )
975
+ assert False , "Failed due to errors.json being created"
972
976
assert "report.html" in files , "Failed to generate report"
973
977
assert (
974
978
"forecast.csv" not in files
@@ -984,6 +988,7 @@ def test_generate_files(operator_setup, model):
984
988
), "Generated metrics file, but `generate_explanation_files` was set False"
985
989
assert not results .get_forecast ().empty
986
990
assert not results .get_metrics ().empty
991
+ print (f"global expl: { results .get_global_explanations ()} " )
987
992
assert not results .get_global_explanations ().empty
988
993
assert not results .get_local_explanations ().empty
989
994
@@ -993,6 +998,10 @@ def test_generate_files(operator_setup, model):
993
998
operator_config = ForecastOperatorConfig .from_dict (yaml_i )
994
999
results = operate (operator_config )
995
1000
files = os .listdir (operator_setup )
1001
+ if "errors.json" in files :
1002
+ with open (os .path .join (operator_setup , "errors.json" )) as f :
1003
+ print (f"Errors in build! { f .read ()} " )
1004
+ assert False , "Failed due to errors.json being created"
996
1005
assert "report.html" in files , "Failed to generate report"
997
1006
assert "forecast.csv" in files , "Failed to generate forecast file"
998
1007
assert "metrics.csv" in files , "Failed to generated metrics file"
0 commit comments