FIX: recompute input features was commented

MeteoSwiss · Nov 7, 2024 · b8c22f0 · b8c22f0
1 parent 52a781e
commit b8c22f0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/rainforest/ml/rf.py b/rainforest/ml/rf.py
@@ -77,9 +77,9 @@ def __init__(self, db_location, input_location=None,
 
         if not valid :
             logging.info('Could not find valid input data from the folder {:s}'.format(input_location))
-        # if force_regenerate_input or not valid:
-        #     logging.info('The program will now compute this input data from the database, this takes quite some time')
-        #     self.prepare_input()
+        if force_regenerate_input or not valid:
+            logging.info('The program will now compute this input data from the database, this takes quite some time')
+            self.prepare_input()
 
     def prepare_input(self, only_center=True, foldername_radar='radar'):
         """

diff --git a/rainforest/ml/rf_train.py b/rainforest/ml/rf_train.py
@@ -149,7 +149,7 @@ def main():
     input_location = str(Path(options.dbfolder, 'rf_input_data'))
     logging.info('Assuming that input data for RF training is in folder {:s}'.format(input_location))
     logging.info('If not available in this folder, they will be computed and stored there')
-    
+
     rf = RFTraining(options.dbfolder, options.inputfolder,
                     options.generate_inputs)