Set default offset to an Offset object (#180)

* Set default offset to an Offset object (fix #174)

* Adapt default offset

* Fix import issue caused by ruff

Co-authored-by: Steffen Schneider <[email protected]>

Author: Ícaro

cebra/data/datasets.py

@@ -30,6 +30,7 @@
 
 import cebra.data as cebra_data
 import cebra.helper as cebra_helper
+from cebra.data.datatypes import Offset
 
 
 class TensorDataset(cebra_data.SingleSessionDataset):
@@ -65,7 +66,7 @@ def __init__(self,
                  neural: Union[torch.Tensor, npt.NDArray],
                  continuous: Union[torch.Tensor, npt.NDArray] = None,
                  discrete: Union[torch.Tensor, npt.NDArray] = None,
-                 offset: int = 1,
+                 offset: Offset = Offset(0, 1),
                  device: str = "cpu"):
         super().__init__(device=device)
         self.neural = self._to_tensor(neural, check_dtype="float").float()