dsipts 1.1.20__tar.gz → 1.1.22__tar.gz

{dsipts-1.1.20 → dsipts-1.1.22}/LONG_DESCRIPTION.md

@@ -398,6 +398,10 @@ and then open the url (http://127.0.0.1:43800)[http://127.0.0.1:43800]. It will
 
 
 
+## 1.1.21-22 (2026-02-02)
+- fixing lightening version support (not yet solved for version >2.6)
+- removed aim (compatibility)
+
 ## 1.1.20 (2026-02-01)
 - automation of tags, version etc using only the makefile
 - update newer version of python

{dsipts-1.1.20 → dsipts-1.1.22}/PKG-INFO

@@ -1,13 +1,12 @@
 Metadata-Version: 2.4
 Name: dsipts
-Version: 1.1.20
+Version: 1.1.22
 Summary: Unified library for timeseries modelling
 Author-email: Andrea Gobbi <agobbi@fbk.eu>
 Project-URL: Homepage, https://github.com/DSIP-FBK/DSIPTS
 Project-URL: Docs, https://dsip-fbk.github.io/DSIPTS/
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
-Requires-Dist: aim
 Requires-Dist: beautifulsoup4==4.12.0
 Requires-Dist: bumpver>=2025.1131
 Requires-Dist: einops>=0.8.1
@@ -18,7 +17,7 @@ Requires-Dist: hydra-joblib-launcher>=1.2.0
 Requires-Dist: hydra-optuna-sweeper>=1.2.0
 Requires-Dist: hydra-submitit-launcher>=1.2.0
 Requires-Dist: ipykernel>=6.30.1
-Requires-Dist: lightning>=2.5.4
+Requires-Dist: lightning==2.5.5
 Requires-Dist: matplotlib>=3.10.6
 Requires-Dist: nbformat>=5.10.4
 Requires-Dist: numba>=0.61.2
@@ -431,6 +430,10 @@ and then open the url (http://127.0.0.1:43800)[http://127.0.0.1:43800]. It will
 
 
 
+## 1.1.21-22 (2026-02-02)
+- fixing lightening version support (not yet solved for version >2.6)
+- removed aim (compatibility)
+
 ## 1.1.20 (2026-02-01)
 - automation of tags, version etc using only the makefile
 - update newer version of python

{dsipts-1.1.20 → dsipts-1.1.22}/pyproject.toml

@@ -1,12 +1,12 @@
 
 [project]
 name = "dsipts"
-version = "1.1.20"
+version = "1.1.22"
 description = "Unified library for timeseries modelling"
 readme = { file = "LONG_DESCRIPTION.md", content-type = "text/markdown" }
 requires-python = ">=3.12"
 dependencies = [
-    "aim",
+    #"aim",
     "beautifulsoup4==4.12.0",
     "bumpver>=2025.1131",
     "einops>=0.8.1",
@@ -17,7 +17,7 @@ dependencies = [
     "hydra-optuna-sweeper>=1.2.0",
     "hydra-submitit-launcher>=1.2.0",
     "ipykernel>=6.30.1",
-    "lightning>=2.5.4",
+    "lightning==2.5.5",
     "matplotlib>=3.10.6",
     "nbformat>=5.10.4",
     "numba>=0.61.2",
@@ -40,7 +40,7 @@ Homepage = "https://github.com/DSIP-FBK/DSIPTS"
 Docs = "https://dsip-fbk.github.io/DSIPTS/"
 
 [tool.bumpver]
-current_version = "1.1.20"
+current_version = "1.1.22"
 version_pattern = "MAJOR.MINOR.PATCH"
 commit = false
 tag = false

{dsipts-1.1.20 → dsipts-1.1.22}/src/dsipts/data_structure/data_structure.py

@@ -32,7 +32,10 @@ from datetime import datetime
 from ..models.utils import weight_init_zeros,weight_init
 import logging 
 from .modifiers import *
-from aim.pytorch_lightning import AimLogger
+disable_aim = True
+
+if disable_aim is False:
+    from aim.pytorch_lightning import AimLogger
 import time
 debug_prediction = False
 class DummyScaler():
@@ -774,6 +777,7 @@ class TimeSeries():
             train_dl = DataLoader(train, batch_size = batch_size , shuffle=True,drop_last=True,num_workers=num_workers,persistent_workers=persistent_workers)
         valid_dl = DataLoader(validation, batch_size = batch_size , shuffle=False,drop_last=True,num_workers=num_workers,persistent_workers=persistent_workers)
         
+        beauty_string(f'train:{len(train_dl)}, validation:{len(valid_dl)}','section',self.verbose)
 
         if debug_prediction:
             dl = DataLoader(test, batch_size = batch_size , shuffle=False,drop_last=True,num_workers=num_workers,persistent_workers=persistent_workers)
@@ -801,11 +805,8 @@ class TimeSeries():
         
         
         #logger = CSVLogger("logs", name=dirpath)
-        aim_logger = AimLogger(
-            experiment=self.name,
-            train_metric_prefix='train_',
-            val_metric_prefix='val_',
-        )
+        beauty_string(f'Init aim','section',self.verbose)
+
 
         #https://stackoverflow.com/questions/49201236/check-the-total-number-of-parameters-in-a-pytorch-model
         n_params = sum(dict((p.data_ptr(), p.numel()) for p in self.model.parameters()).values())
@@ -819,17 +820,25 @@ class TimeSeries():
             buffer_size += buffer.nelement() * buffer.element_size()
 
         size_all_mb = (param_size + buffer_size) / 1024**2
-        #aim_logger.experiment.track(self.model.name,name='model_name')
-
-        aim_logger.experiment.track(n_params,name='N-parameters')
-        aim_logger.experiment.track(size_all_mb,name='dim-model-MB')
-        aim_logger.experiment.track(len(train_dl.dataset),name='len-train')
-        aim_logger.experiment.track(len(valid_dl.dataset),name='len-valid')
-        #aim_logger.experiment.track(self.config,name=None)
         tmp = self.config.copy()
         tmp['model_name'] = self.model.name
-        aim_logger._run['hyperparameters'] = tmp
 
+        #aim_logger.experiment.track(self.model.name,name='model_name')
+        if disable_aim is False:
+            aim_logger = AimLogger(
+                experiment=self.name,
+                train_metric_prefix='train_',
+                val_metric_prefix='val_',
+            )
+            aim_logger.experiment.track(n_params,name='N-parameters')
+            aim_logger.experiment.track(size_all_mb,name='dim-model-MB')
+            aim_logger.experiment.track(len(train_dl.dataset),name='len-train')
+            aim_logger.experiment.track(len(valid_dl.dataset),name='len-valid')
+            #aim_logger.experiment.track(self.config,name=None)
+            
+            aim_logger._run['hyperparameters'] = tmp
+        else:
+            aim_logger = None
         mc = MetricsCallback(dirpath)
         ## TODO se ci sono 2 o piu gpu MetricsCallback non funziona (secondo me fa una istanza per ogni dataparallel che lancia e poi non riesce a recuperare info)
         pl.seed_everything(seed, workers=True)
@@ -884,7 +893,8 @@ class TimeSeries():
         tot_seconds = time.time()
 
 
-      
+        beauty_string(f'tuning lr','section',self.verbose)
+
      
         if auto_lr_find and (weight_exists is False):
             try:
@@ -902,7 +912,8 @@ class TimeSeries():
                     self.model.optim_config['lr'] = lr_finder.suggestion() ## we are using it as optim key
             except Exception as e:
                 beauty_string(f'There is a problem with the finding LR routine {e}','section',self.verbose)
-        
+        beauty_string(f'Start fitting','section',self.verbose)
+
         if OLD_PL:
             if weight_exists:
                 trainer.fit(self.model, train_dl,valid_dl,ckpt_path=os.path.join(dirpath,'last.ckpt'))
@@ -994,8 +1005,10 @@ class TimeSeries():
         
         beauty_string('END of the training process','block',self.verbose)
 
-        aim_logger.experiment.track((time.time()-tot_seconds),name='seconds-training')
-        aim_logger.experiment.track(val_loss,name='val-loss-end-train')
+
+        if disable_aim is False:
+            aim_logger.experiment.track((time.time()-tot_seconds),name='seconds-training')
+            aim_logger.experiment.track(val_loss,name='val-loss-end-train')
 
         
         

{dsipts-1.1.20 → dsipts-1.1.22}/src/dsipts/models/base_v2.py

@@ -360,7 +360,10 @@ class Base(pl.LightningModule):
                 ax.plot(pred,'o-',label='pred')
                 ax.legend()
                 ax.set_title(f'Channel {i} first element first batch validation {int(100*self.count_epoch/self.trainer.max_epochs)}%')
-                self.logger.experiment.track(Image(fig), name='cm_training_end')
+                try:
+                    self.logger.experiment.track(Image(fig), name='cm_training_end')
+                except:
+                    pass ##no aim probably
                 #self.log(f"example_{i}", np.stack([real, pred]).T,sync_dist=True)
                 plt.close(fig) 
             

dsipts-1.1.22/src/dsipts/version.py

@@ -0,0 +1 @@
+__version__ = "1.1.22"

{dsipts-1.1.20 → dsipts-1.1.22}/src/dsipts.egg-info/PKG-INFO

@@ -1,13 +1,12 @@
 Metadata-Version: 2.4
 Name: dsipts
-Version: 1.1.20
+Version: 1.1.22
 Summary: Unified library for timeseries modelling
 Author-email: Andrea Gobbi <agobbi@fbk.eu>
 Project-URL: Homepage, https://github.com/DSIP-FBK/DSIPTS
 Project-URL: Docs, https://dsip-fbk.github.io/DSIPTS/
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
-Requires-Dist: aim
 Requires-Dist: beautifulsoup4==4.12.0
 Requires-Dist: bumpver>=2025.1131
 Requires-Dist: einops>=0.8.1
@@ -18,7 +17,7 @@ Requires-Dist: hydra-joblib-launcher>=1.2.0
 Requires-Dist: hydra-optuna-sweeper>=1.2.0
 Requires-Dist: hydra-submitit-launcher>=1.2.0
 Requires-Dist: ipykernel>=6.30.1
-Requires-Dist: lightning>=2.5.4
+Requires-Dist: lightning==2.5.5
 Requires-Dist: matplotlib>=3.10.6
 Requires-Dist: nbformat>=5.10.4
 Requires-Dist: numba>=0.61.2
@@ -431,6 +430,10 @@ and then open the url (http://127.0.0.1:43800)[http://127.0.0.1:43800]. It will
 
 
 
+## 1.1.21-22 (2026-02-02)
+- fixing lightening version support (not yet solved for version >2.6)
+- removed aim (compatibility)
+
 ## 1.1.20 (2026-02-01)
 - automation of tags, version etc using only the makefile
 - update newer version of python

{dsipts-1.1.20 → dsipts-1.1.22}/src/dsipts.egg-info/requires.txt

@@ -1,4 +1,3 @@
-aim
 beautifulsoup4==4.12.0
 bumpver>=2025.1131
 einops>=0.8.1
@@ -9,7 +8,7 @@ hydra-joblib-launcher>=1.2.0
 hydra-optuna-sweeper>=1.2.0
 hydra-submitit-launcher>=1.2.0
 ipykernel>=6.30.1
-lightning>=2.5.4
+lightning==2.5.5
 matplotlib>=3.10.6
 nbformat>=5.10.4
 numba>=0.61.2

dsipts-1.1.20/src/dsipts/version.py

@@ -1 +0,0 @@
-__version__ = "1.1.20"