fix test

ibm-granite · Oct 25, 2024 · 3e1f4a5 · 3e1f4a5
1 parent 420d53f
commit 3e1f4a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/services/inference/tests/test_inference.py b/services/inference/tests/test_inference.py
@@ -135,7 +135,7 @@ def test_zero_shot_forecast_inference(ts_data):
         },
         "schema": {
             "timestamp_column": params["timestamp_column"],
-            # "id_columns": params["id_columns"],
+            "id_columns": params["id_columns"],
             "target_columns": params["target_columns"],
         },
         "data": encode_data(test_data_.iloc[:2], params["timestamp_column"]),