Merge pull request #21 from tamnva/development

fix: load data.pt file with pytorch

Author: tamnva

hydroecolstm/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.3.6"
+__version__ = "0.3.7"
 __author__ = 'Tam V. Nguyen'
 __credits__ = 'Helmholtz Centre for Environmental Research (UFZ)'
 

hydroecolstm/interface/visualize_frame.py

@@ -421,9 +421,8 @@ def load_data_event(self):
             file_name=ctk.filedialog.askopenfilename(title="Select data.pt file type", 
                                                      filetypes=(('pt files', '*.pt'),
                                                                 ('All files', '*.*')))
-            self.globalData.update(torch.load(file_name))
+            self.globalData.update(torch.load(file_name, weights_only=False))
             self.selected_data_file.configure(text= '...' + file_name[-30:])
-            print(self.globalData.keys())
         except:
             pass
 

hydroecolstm/model_run.py

@@ -53,6 +53,7 @@ def run_config(config):
         # Create model 
         best_result = result.get_best_result("loss", mode="min")
         best_config = best_result.config
+        
         del best_result.config["x_train_scale"], 
         del best_result.config["y_train_scale"]
         del best_result.config["x_valid_scale"]

hydroecolstm/train/trainer.py

@@ -134,7 +134,7 @@ def train(self,
                                        check_point)
 
             if early_stopping.early_stop:
-                print("Early stopping")
+                print("Early stopping.")
                 break
 
         # If the model does not stops until the last epoch

setup.py

@@ -8,7 +8,7 @@
 
 setup(
     name='hydroecolstm',
-    version='0.3.6',    
+    version='0.3.7',    
     description='A python package for HydroEcological Modelling using LSTM',
     long_description=long_description,
     long_description_content_type='text/markdown',