[FIX] ADD forecasting init design to pip data files (#459)

* add forecasting_init.json to data files under setup * avoid undefined reference in scale_value
automl · Aug 5, 2022 · c7220f7 · c7220f7
1 parent 0e574af
commit c7220f7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/autoPyTorch/pipeline/components/setup/network/forecasting_architecture.py b/autoPyTorch/pipeline/components/setup/network/forecasting_architecture.py
@@ -368,7 +368,9 @@ def scale_value(self,
                 outputs = raw_value - loc.to(device)
             else:
                 outputs = (raw_value - loc.to(device)) / scale.to(device)
-        return outputs
+            return outputs
+        else:
+            return raw_value
 
     @abstractmethod
     def forward(self,

diff --git a/setup.py b/setup.py
@@ -83,6 +83,7 @@
     },
     test_suite="pytest",
     data_files=[('configs', ['autoPyTorch/configs/default_pipeline_options.json']),
-                ('portfolio', ['autoPyTorch/configs/greedy_portfolio.json'])],
+                ('portfolio', ['autoPyTorch/configs/greedy_portfolio.json']),
+                ('forecasting_init', ['autoPyTorch/configs/forecasting_init_cfgs.json'])],
     dependency_links=['https://github.com/automl/automl_common.git/tarball/autoPyTorch#egg=package-0.0.1']
 )