AdaptiveMotorControlLab / CEBRA

Learnable latent embeddings for joint behavioral and neural analysis - Official implementation of CEBRA
https://cebra.ai
Other
884 stars 72 forks source link

Fix dtypes handled in the labels of the decoders #30

Closed CeliaBenquet closed 1 year ago

CeliaBenquet commented 1 year ago

int and float subtypes accepted to be in the labels of the decoders were hard coded, which results in errors for int8 for instance.

Now, it supports all subtypes of Integer and floating in numpy arrays and pytorch tensors.

Fix https://github.com/AdaptiveMotorControlLab/CEBRA-dev/pull/616

stes commented 1 year ago

Seems like the documentation status still fails due to

image

I'll disable the comment function for now so we can merge this and other upcoming PRs, the check is still executed.