Skip to content

Commit 5b6d50d

Browse files
committed
updated test cases
1 parent 4f91041 commit 5b6d50d

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tests/operators/anomaly/test_anomaly_simple.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
# Copyright (c) 2023, 2024 Oracle and/or its affiliates.
44
# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
55

6-
from ads.opctl.operator.lowcode.anomaly.const import NonTimeADSupportedModels
6+
from ads.opctl.operator.lowcode.anomaly.const import NonTimeADSupportedModels, SupportedModels
77
import yaml
88
import subprocess
99
import pandas as pd
@@ -16,8 +16,7 @@
1616
from datetime import datetime
1717
from ads.opctl.operator.cmd import run
1818

19-
20-
MODELS = ["autots"] # "automlx",
19+
MODELS = ["autots", "iqr", "lof", "zscore", "rolling_zscore", "mad", "ee", "isolationforest"]
2120

2221
# Mandatory YAML parameters
2322
TEMPLATE_YAML = {
@@ -218,7 +217,7 @@ def test_load_datasets(model, data_dict):
218217
yaml_i = deepcopy(TEMPLATE_YAML)
219218
yaml_i["spec"]["model"] = model
220219
yaml_i["spec"]["input_data"]["url"] = data_dict["url"]
221-
if model in NonTimeADSupportedModels.values():
220+
if model in set(NonTimeADSupportedModels.values()) - set(SupportedModels.values()):
222221
del yaml_i["spec"]["datetime_column"]
223222
else:
224223
yaml_i["spec"]["datetime_column"]["name"] = data_dict["dt_col"]

0 commit comments

Comments
 (0)