bulk update

databricks-industry-solutions · May 16, 2024 · 6763e15 · 6763e15
1 parent dcc3101
commit 6763e15
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/RUNME.py b/RUNME.py
@@ -70,7 +70,9 @@
                 },
                 "num_workers": 0,
                 "node_type_id": {"AWS": "i3.xlarge", "MSA": "Standard_DS3_v2", "GCP": "n1-highmem-4"},
+                "driver_node_type_id": {"AWS": "i3.xlarge", "MSA": "Standard_DS3_v2", "GCP": "n1-highmem-4"},
                 "custom_tags": {
+                    "ResourceClass": "SingleNode",
                     "usage": "solacc_testing"
                 },
             }

diff --git a/forecasting_sa/models/sktime/SKTimeForecastingPipeline.py b/forecasting_sa/models/sktime/SKTimeForecastingPipeline.py
@@ -61,6 +61,9 @@ def fit(self, x, y=None):
 
     def predict(self, hist_df: pd.DataFrame, val_df: pd.DataFrame = None):
         _df = self.prepare_data(hist_df)
+
+        print(f"_df: {_df}")
+
         self.fit(_df)
         pred_df = self.model.predict(
             ForecastingHorizon(np.arange(1, self.params.prediction_length + 1))