Merge branch 'main' into celia/fix-790-public

AdaptiveMotorControlLab · Aug 24, 2024 · d83986c · d83986c
2 parents 2d5b0b5 + 65d10a2
commit d83986c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/cebra/data/datasets.py b/cebra/data/datasets.py
@@ -73,6 +73,7 @@ def __init__(
         continuous: Union[torch.Tensor, npt.NDArray] = None,
         discrete: Union[torch.Tensor, npt.NDArray] = None,
         offset: int = 1,
+        device: str = "cpu"
     ):
         super().__init__(device=device)
         self.neural = self._to_tensor(neural, check_dtype="float").float()

diff --git a/docs/source/installation.rst b/docs/source/installation.rst
@@ -54,8 +54,9 @@ we outline different options below.
 
         .. code:: bash
 
-            $ conda create -n cebra python==3.8
+            $ conda create -n cebra python==3.12
             $ conda activate cebra
+            $ conda install -c conda-forge pytables==3.8.0
 
         .. rubric:: Install PyTorch separately