Release 0.6.1 (#294)

* Update download urls from figshare to s3 bucket

* minimize the diff

* unify the dataset download funcs

* fix accedentally skipped save/load test

* include a numpy legacy test

* fix windows compatibility for tempfile

* fix added test

* Fix legacy loading logic

* minimize diff in tests

* Fix _assert_equal check

* Bump version to 0.6.1

* fix loading logic for legacy torch

* allowlist float32d

* bump python and torch in workflows

* add torch 2.6.0 as legacy only

* fix codespell typo

Author: stes

.github/workflows/build.yml

@@ -23,11 +23,11 @@ jobs:
         #   https://devguide.python.org/versions/
         # Python Ubuntu LTS:
         #   https://documentation.ubuntu.com/ubuntu-for-developers/reference/availability/python/
-        python-version: ["3.10", "3.12", "3.13"]
+        python-version: ["3.10", "3.12", "3.13", "3.14"]
         # We aim to support the versions on pytorch.org
         # as well as selected previous versions on
         # https://pytorch.org/get-started/previous-versions/
-        torch-version: ["2.6.0", "2.10.0"]
+        torch-version: ["2.9.1", "2.12.0"]
         sklearn-version: ["latest"]
         numpy-version: ["latest"]
 
@@ -55,10 +55,10 @@ jobs:
           # TODO(stes): latest torch and python
           # requires a PyTables release compatible with
           # python 3.14. Update when new version is released.
-          #- os: ubuntu-latest
-          #  torch-version: 2.9.1
-          #  python-version: "3.14"
-          #  sklearn-version: "latest"
+          - os: ubuntu-latest
+            torch-version: 2.9.1
+            python-version: "3.14"
+            sklearn-version: "latest"
 
           # legacy support
           - os: ubuntu-latest

Dockerfile

@@ -40,7 +40,7 @@ RUN make dist
 FROM cebra-base
 
 # install the cebra wheel
-ENV WHEEL=cebra-0.6.0-py3-none-any.whl
+ENV WHEEL=cebra-0.6.1-py3-none-any.whl
 WORKDIR /build
 COPY --from=wheel /build/dist/${WHEEL} .
 RUN pip install --no-cache-dir ${WHEEL}'[dev,integrations,datasets]'

Makefile

@@ -1,4 +1,4 @@
-CEBRA_VERSION := 0.6.0
+CEBRA_VERSION := 0.6.1
 
 dist:
 	python3 -m pip install virtualenv

PKGBUILD

@@ -1,7 +1,7 @@
 # Maintainer: Steffen Schneider <stes@hey.com>
 pkgname=python-cebra
 _pkgname=cebra
-pkgver=0.6.0
+pkgver=0.6.1
 pkgrel=1
 pkgdesc="Consistent Embeddings of high-dimensional Recordings using Auxiliary variables"
 url="https://cebra.ai"

cebra/__init__.py

@@ -66,7 +66,7 @@
 
 import cebra.integrations.sklearn as sklearn
 
-__version__ = "0.6.0"
+__version__ = "0.6.1"
 __all__ = ["CEBRA"]
 __allow_lazy_imports = False
 __lazy_imports = {}

cebra/distributions/index.py

@@ -215,7 +215,7 @@ def search(self, continuous, discrete=None):
                 Samples from the continuous index
             discrete:
                 Optionally matching samples from the discrete index,
-                used to pre-select matching indices.
+                used to preselect matching indices.
         """
         if continuous.shape[1] != self.continuous.shape[1]:
             raise ValueError(f"Shape of continuous index does not match along "

reinstall.sh

@@ -15,7 +15,7 @@ pip uninstall -y cebra
 # Get version info after uninstalling --- this will automatically get the
 # most recent version based on the source code in the current directory.
 # $(tools/get_cebra_version.sh)
-VERSION=0.6.0
+VERSION=0.6.1
 echo "Upgrading to CEBRA v${VERSION}"
 
 # Upgrade the build system (PEP517/518 compatible)