build fix

Author: prasankh

tests/operators/forecast/test_errors.py

@@ -869,17 +869,22 @@ def test_what_if_analysis(operator_setup, model):
 pytest.skip("Skipping what-if scenario for auto-select")
 tmpdirname = operator_setup
 historical_data_path, additional_data_path = setup_small_rossman()
-additional_path = f'{tmpdirname}/additional_data.csv'
-additional_data = pd.read_csv(additional_data_path)
-numeric_columns = additional_data.select_dtypes(include=['number', 'object'])
+additional_test_path = f'{tmpdirname}/additional_data.csv'
+historical_test_path = f'{tmpdirname}/historical_data.csv'
+historical_data = pd.read_csv(historical_data_path, parse_dates=["Date"])
+historical_filtered = historical_data[historical_data['Date'] > "2013-03-01"]
+additional_data = pd.read_csv(additional_data_path, parse_dates=["Date"])
+add_filtered = additional_data[additional_data['Date'] > "2013-03-01"]
+numeric_columns = add_filtered.select_dtypes(include=['number', 'object', 'datetime64'])
 non_constant_columns = numeric_columns.columns[(numeric_columns != numeric_columns.iloc[0]).any()]
 df_non_constant = numeric_columns[non_constant_columns.union(['Store'])]
-df_non_constant.to_csv(f'{additional_path}', index=False)
+df_non_constant.to_csv(f'{additional_test_path}', index=False)
+historical_filtered.to_csv(f'{historical_test_path}', index=False)
 
 yaml_i, output_data_path = populate_yaml(
 tmpdirname=tmpdirname,
-historical_data_path=historical_data_path,
-additional_data_path=additional_path,
+historical_data_path=historical_test_path,
+additional_data_path=additional_test_path,
 output_data_path=f"{tmpdirname}/{model}/results"
 )
 yaml_i["spec"]["horizon"] = 10