Merge branch 'main' into unified-cebra

Author: MMathisLab

cebra/datasets/demo.py

@@ -36,6 +36,7 @@
 NUMS_NEURAL = [3, 4, 5]
 
 
+
 class DemoDataset(cebra.data.SingleSessionDataset):
 
     def __init__(self, num_timepoints=_DEFAULT_NUM_TIMEPOINTS, num_neural=4):

cebra/solver/base.py

@@ -396,7 +396,6 @@ def num_parameters(self) -> int:
     @abc.abstractmethod
     def parameters(self, session_id: Optional[int] = None):
         """Iterate over all parameters of the model.
-
         Args:
             session_id: The session ID, an :py:class:`int` between 0 and
                 the number of sessions -1 for multisession, and set to

cebra/solver/single_session.py

@@ -24,6 +24,7 @@
 import copy
 from typing import Optional, Tuple
 
+
 import literate_dataclasses as dataclasses
 import torch
 

tests/test_solver.py

@@ -41,10 +41,8 @@ def _get_loader(data_name, loader_initfunc):
     loader = loader_initfunc(data, **kwargs)
     return loader, data
 
-
 OUTPUT_DIMENSION = 3
 
-
 def _make_model(dataset, model_architecture="offset10-model"):
     # TODO flexible input dimension
     # return nn.Sequential(
@@ -400,4 +398,4 @@ def test_unified_session(data_name, model_architecture, loader_initfunc,
         assert emb.shape == (loader.dataset.num_timepoints, 3)
 
         emb = solver.transform(data, labels, session_id=i, batch_size=300)
-        assert emb.shape == (loader.dataset.num_timepoints, 3)
+        assert emb.shape == (loader.dataset.num_timepoints, 3)