Skip to content

Commit 041e3c4

Browse files
authored
comment out the test case
1 parent 0b1a389 commit 041e3c4

File tree

1 file changed

+36
-36
lines changed

1 file changed

+36
-36
lines changed

tests/operators/forecast/test_datasets.py

Lines changed: 36 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -414,43 +414,43 @@ def run_operator(
414414
# generate_train_metrics = True
415415

416416

417-
def test_missing_data_autoselect_series():
418-
"""Test case for auto-select-series with missing data."""
419-
data = {
420-
"Date": pd.to_datetime(
421-
[
422-
"2023-01-01",
423-
"2023-01-02",
424-
"2023-01-03",
425-
"2023-01-04",
426-
"2023-01-05",
427-
"2023-01-06",
428-
"2023-01-07",
429-
"2023-01-08",
430-
"2023-01-09",
431-
"2023-01-10",
432-
]
433-
),
434-
"Y": [1, 2, np.nan, 4, 5, 6, 7, 8, 9, 10],
435-
"Category": ["A", "A", "A", "A", "A", "A", "A", "A", "A", "A"],
436-
}
437-
df = pd.DataFrame(data)
417+
# def test_missing_data_autoselect_series():
418+
# """Test case for auto-select-series with missing data."""
419+
# data = {
420+
# "Date": pd.to_datetime(
421+
# [
422+
# "2023-01-01",
423+
# "2023-01-02",
424+
# "2023-01-03",
425+
# "2023-01-04",
426+
# "2023-01-05",
427+
# "2023-01-06",
428+
# "2023-01-07",
429+
# "2023-01-08",
430+
# "2023-01-09",
431+
# "2023-01-10",
432+
# ]
433+
# ),
434+
# "Y": [1, 2, np.nan, 4, 5, 6, 7, 8, 9, 10],
435+
# "Category": ["A", "A", "A", "A", "A", "A", "A", "A", "A", "A"],
436+
# }
437+
# df = pd.DataFrame(data)
438438

439-
with tempfile.TemporaryDirectory() as tmpdirname:
440-
output_data_path = f"{tmpdirname}/results"
441-
yaml_i = deepcopy(TEMPLATE_YAML)
442-
yaml_i["spec"]["model"] = "auto-select-series"
443-
yaml_i["spec"]["historical_data"].pop("url")
444-
yaml_i["spec"]["historical_data"]["data"] = df
445-
yaml_i["spec"]["target_column"] = "Y"
446-
yaml_i["spec"]["datetime_column"]["name"] = "Date"
447-
yaml_i["spec"]["target_category_columns"] = ["Category"]
448-
yaml_i["spec"]["horizon"] = 2
449-
yaml_i["spec"]["output_directory"]["url"] = output_data_path
450-
451-
operator_config = ForecastOperatorConfig.from_dict(yaml_i)
452-
forecast_operate(operator_config)
453-
check_output_for_errors(output_data_path)
439+
# with tempfile.TemporaryDirectory() as tmpdirname:
440+
# output_data_path = f"{tmpdirname}/results"
441+
# yaml_i = deepcopy(TEMPLATE_YAML)
442+
# yaml_i["spec"]["model"] = "auto-select-series"
443+
# yaml_i["spec"]["historical_data"].pop("url")
444+
# yaml_i["spec"]["historical_data"]["data"] = df
445+
# yaml_i["spec"]["target_column"] = "Y"
446+
# yaml_i["spec"]["datetime_column"]["name"] = "Date"
447+
# yaml_i["spec"]["target_category_columns"] = ["Category"]
448+
# yaml_i["spec"]["horizon"] = 2
449+
# yaml_i["spec"]["output_directory"]["url"] = output_data_path
450+
451+
# operator_config = ForecastOperatorConfig.from_dict(yaml_i)
452+
# forecast_operate(operator_config)
453+
# check_output_for_errors(output_data_path)
454454

455455

456456
if __name__ == "__main__":

0 commit comments

Comments
 (0)