nit

databricks · Oct 7, 2024 · 0930423 · 0930423
1 parent ac27c97
commit 0930423
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/runtime/tests/automl_runtime/forecast/deepar/model_test.py b/runtime/tests/automl_runtime/forecast/deepar/model_test.py
@@ -108,7 +108,7 @@ def test_model_save_and_load_single_series(self):
         run_id = run.info.run_id
 
         # check if all additional dependencies are logged
-        self.check_requirements(run_id)
+        self._check_requirements(run_id)
 
         # load the model and predict
         loaded_model = mlflow.pyfunc.load_model(f"runs:/{run_id}/model")