File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -738,10 +738,11 @@ def test_pandas_historical_input(operator_setup, model):
738738 ForecastOperatorConfig ,
739739 )
740740
741- tmpdirname = operator_setup
742- historical_data_path , additional_data_path = setup_artificial_data ()
741+ historical_data_path , additional_data_path , _ = setup_artificial_data (
742+ operator_setup
743+ )
743744 yaml_i , output_data_path = populate_yaml (
744- tmpdirname = tmpdirname ,
745+ operator_setup = operator_setup ,
745746 historical_data_path = historical_data_path ,
746747 additional_data_path = additional_data_path ,
747748 )
@@ -755,7 +756,7 @@ def test_pandas_historical_input(operator_setup, model):
755756 operator_config = ForecastOperatorConfig .from_dict (yaml_i )
756757 operate (operator_config )
757758 assert pd .read_csv (additional_data_path )["Date" ].equals (
758- pd .read_csv (f"{ tmpdirname } /results/forecast.csv" )["Date" ]
759+ pd .read_csv (f"{ operator_setup } /results/forecast.csv" )["Date" ]
759760 )
760761
761762
You can’t perform that action at this time.
0 commit comments