Add target col back to required cols in DeepAR predict() (#150)

databricks · Oct 1, 2024 · f7dbeb3 · f7dbeb3
1 parent 4f16567
commit f7dbeb3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/runtime/databricks/automl_runtime/forecast/deepar/model.py b/runtime/databricks/automl_runtime/forecast/deepar/model.py
@@ -77,7 +77,7 @@ def predict(self,
         :return: predicted pd.DataFrame that starts after the last timestamp in the input dataframe,
                                 and predicts the horizon using the mean of the samples
         """
-        required_cols = [self._time_col]
+        required_cols = [self._target_col, self._time_col]
         if self._id_cols:
             required_cols += self._id_cols
         self._validate_cols(model_input, required_cols)

diff --git a/runtime/databricks/automl_runtime/version.py b/runtime/databricks/automl_runtime/version.py
@@ -14,4 +14,4 @@
 # limitations under the License.
 #
 
-__version__ = "0.2.20.2"  # pragma: no cover
+__version__ = "0.2.20.3.dev"  # pragma: no cover