Fix tests for upgraded dependencies (#184)

Author: stes

cebra/datasets/monkey_reaching.py

@@ -77,8 +77,12 @@ def _load_data(
             "Could not import the nlb_tools package required for data loading "
             "the raw reaching datasets in NWB format. "
             "If required, you can install the dataset by running "
-            "pip install nlb_tools or installing cebra with the [datasets] "
-            "dependencies: pip install 'cebra[datasets]'")
+            "pip install nlb_tools."
+            # NOTE(stes): Install nlb_tools manually for now, also see
+            # note in setup.cfg
+            # or installing cebra with the [datasets] "
+            #"dependencies: pip install 'cebra[datasets]'")
+        )
 
     def _get_info(trial_info, data):
         passive = []

cebra/integrations/plotly.py

@@ -45,7 +45,7 @@ def _convert_cmap2colorscale(cmap: str, pl_entries: int = 11, rdigits: int = 2):
     """
     scale = np.linspace(0, 1, pl_entries)
     colors = (cmap(scale)[:, :3] * 255).astype(np.uint8)
-    pl_colorscale = [[round(s, rdigits), f"rgb{tuple(color)}"]
+    pl_colorscale = [[float(round(s, rdigits)), f"rgb{tuple(color.tolist())}"]
                      for s, color in zip(scale, colors)]
     return pl_colorscale
 

setup.cfg

@@ -31,6 +31,7 @@ where =
 python_requires = >=3.9
 install_requires =
     joblib
+    numpy<2.0.0
     literate-dataclasses
     scikit-learn
     scipy
@@ -44,7 +45,13 @@ datasets =
     # cebra.datasets.allen
     h5py
     pandas
-    nlb_tools
+    # NOTE(stes): nlb_tools currently pins pandas to <1.3.4, see here:
+    # https://github.com/neurallatents/nlb_tools/blob/1ddc15f45b56388ff093d1396b7b87b36fa32a68/requirements.txt#L1
+    # Since this is a fairly old pandas version, it causes additional version
+    # conflicts with other packages. Removing nlb_tools as a required dependency for now.
+    # The only part this package is needed is in cebra/datasets/monkey_reaching.py, where
+    # we added a warning message to tell the user how to manually install it.
+    #nlb_tools
     # additional data loading dependencies
     hdf5storage # for creating .mat files in new format
     openpyxl # for excel file format loading