Skip to content

Commit 7fa977b

Browse files
committed
update tests
1 parent fa8c55d commit 7fa977b

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

ads/opctl/operator/lowcode/forecast/model/automlx.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,9 @@ def _build_model(self) -> pd.DataFrame:
113113
data_i = self.drop_horizon(data)
114114
X_pred = self.get_horizon(data).drop(target, axis=1)
115115

116-
logger.debug(f"Time Index Monotonic: {data_i.index.is_monotonic}")
116+
logger.debug(
117+
f"Time Index Monotonic: {data_i.index.is_monotonic_increasing}"
118+
)
117119

118120
if self.loaded_models is not None and s_id in self.loaded_models:
119121
model = self.loaded_models[s_id]

tests/operators/forecast/test_datasets.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -100,10 +100,7 @@ def test_load_datasets(model, data_details):
100100
include_test_data = data_details.get("include_test_data", True)
101101

102102
dataset_i = pd.read_csv(dataset_name)
103-
104103
additional_cols = list(set(dataset_i.columns) - {DATETIME_COL, target})
105-
target_df = dataset_i[target][:-PERIODS]
106-
test_df = dataset_i[target][-PERIODS:]
107104

108105
print(dataset_name, len(target))
109106
with tempfile.TemporaryDirectory() as tmpdirname:

0 commit comments

Comments
 (0)