change Int32 to Int64 (#751)

aeye-lab · Jul 10, 2024 · c83d532 · c83d532
1 parent 1e2eac1
commit c83d532
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 11 deletions.
diff --git a/src/pymovements/datasets/gazebase.py b/src/pymovements/datasets/gazebase.py
@@ -165,9 +165,9 @@ class GazeBase(DatasetDefinition):
                 'n': pl.Int64,
                 'x': pl.Float32,
                 'y': pl.Float32,
-                'val': pl.Int32,
+                'val': pl.Int64,
                 'dP': pl.Float32,
-                'lab': pl.Int32,
+                'lab': pl.Int64,
                 'xT': pl.Float32,
                 'yT': pl.Float32,
             },

diff --git a/src/pymovements/datasets/judo1000.py b/src/pymovements/datasets/judo1000.py
@@ -153,8 +153,8 @@ class JuDo1000(DatasetDefinition):
     custom_read_kwargs: dict[str, Any] = field(
         default_factory=lambda: {
             'dtypes': {
-                'trialId': pl.Int32,
-                'pointId': pl.Int32,
+                'trialId': pl.Int64,
+                'pointId': pl.Int64,
                 'time': pl.Int64,
                 'x_left': pl.Float32,
                 'y_left': pl.Float32,

diff --git a/src/pymovements/datasets/sb_sat.py b/src/pymovements/datasets/sb_sat.py
@@ -145,7 +145,7 @@ class SBSAT(DatasetDefinition):
             'dtypes': {
                 'time': pl.Int64,
                 'book_name': pl.Utf8,
-                'screen_id': pl.Int32,
+                'screen_id': pl.Int64,
                 'x_left': pl.Float32,
                 'y_left': pl.Float32,
                 'pupil_left': pl.Float32,

diff --git a/tests/unit/gaze/transforms/downsample_test.py b/tests/unit/gaze/transforms/downsample_test.py
@@ -62,20 +62,20 @@ def test_downsample_init_raises_error(factor, exception, msg_substrings):
         ),
         pytest.param(
             1,
-            pl.Series('pixel', [1, 2, 3], pl.Int32),
-            pl.Series('pixel', [1, 2, 3], pl.Int32),
+            pl.Series('pixel', [1, 2, 3], pl.Int64),
+            pl.Series('pixel', [1, 2, 3], pl.Int64),
             id='factor_1_returns_same_series',
         ),
         pytest.param(
             2,
-            pl.Series('pixel', [0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5], pl.Int32),
-            pl.Series('pixel', [0, 1, 2, 3, 4, 5], pl.Int32),
+            pl.Series('pixel', [0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5], pl.Int64),
+            pl.Series('pixel', [0, 1, 2, 3, 4, 5], pl.Int64),
             id='factor_2_returns_every_second_item',
         ),
         pytest.param(
             3,
-            pl.Series('pixel', [0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5], pl.Int32),
-            pl.Series('pixel', [0, 1, 2, 3, 4, 5], pl.Int32),
+            pl.Series('pixel', [0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5], pl.Int64),
+            pl.Series('pixel', [0, 1, 2, 3, 4, 5], pl.Int64),
             id='factor_3_returns_every_third_item',
         ),
     ],