Fix circular import and abstract method

Author: CeliaBenquet

cebra/data/datasets.py

@@ -28,11 +28,11 @@
 import numpy.typing as npt
 import torch
 
+import cebra
 import cebra.data as cebra_data
 import cebra.data.masking as cebra_data_masking
 import cebra.helper as cebra_helper
 import cebra.io as cebra_io
-import cebra.models
 from cebra.data.datatypes import Batch
 from cebra.data.datatypes import BatchIndex
 from cebra.data.datatypes import Offset

cebra/solver/base.py

@@ -393,7 +393,6 @@ def num_parameters(self) -> int:
         """Total number of parameters in the encoder and criterion."""
         return sum(p.numel() for p in self.parameters())
 
-    @abc.abstractmethod
     def parameters(self, session_id: Optional[int] = None):
         """Iterate over all parameters of the model.