Fix extra docs errors

Author: CeliaBenquet

cebra/data/multi_session.py

@@ -108,19 +108,7 @@ def configure_for(self, model: "cebra.models.Model"):
         """Configure the dataset offset for the provided model.
 
         Call this function before indexing the dataset. This sets the
-        `offset` attribute of the dataset.
-
-        Args:
-            model: The model to configure the dataset for.
-        """
-        for i, session in enumerate(self.iter_sessions()):
-            session.configure_for(model[i])
-
-    def configure_for(self, model: "cebra.models.Model"):
-        """Configure the dataset offset for the provided model.
-
-        Call this function before indexing the dataset. This sets the
-        :py:attr:`offset` attribute of the dataset.
+        :py:attr:`cebra_data.Dataset.offset` attribute of the dataset.
 
         Args:
             model: The model to configure the dataset for.

cebra/data/single_session.py

@@ -73,7 +73,7 @@ def configure_for(self, model: "cebra.models.Model"):
         """Configure the dataset offset for the provided model.
 
         Call this function before indexing the dataset. This sets the
-        :py:attr:`offset` attribute of the dataset.
+        :py:attr:`cebra_data.Dataset.offset` attribute of the dataset.
 
         Args:
             model: The model to configure the dataset for.

tests/test_solver.py

@@ -68,6 +68,7 @@
     multi_session_tests.append((*args, cebra.solver.MultiSessionSolver))
 
 
+
 def _get_loader(data, loader_initfunc):
     kwargs = dict(num_steps=5, batch_size=32)
     loader = loader_initfunc(data, **kwargs)