hotfix: check whether public dataset has gaze files (#872)

aeye-lab · Oct 24, 2024 · 49802e4 · 49802e4
1 parent 502dc44
commit 49802e4
Showing 1 changed file with 12 additions and 11 deletions.
diff --git a/tests/integration/public_dataset_processing_test.py b/tests/integration/public_dataset_processing_test.py
@@ -39,14 +39,15 @@ def test_public_dataset_processing(dataset_name, tmp_path):
     dataset.load()
 
     # Do some basic transformations.
-    if 'pixel' in dataset.gaze[0].columns:
-        dataset.pix2deg()
-    dataset.pos2vel()
-    dataset.pos2acc()
-
-    for gaze in dataset.gaze:
-        assert 'position' in gaze.columns
-        assert 'velocity' in gaze.columns
-        assert 'acceleration' in gaze.columns
-
-    shutil.rmtree(dataset_path, ignore_errors=True)
+    if dataset.definition.has_files['gaze']:
+        if 'pixel' in dataset.gaze[0].columns:
+            dataset.pix2deg()
+        dataset.pos2vel()
+        dataset.pos2acc()
+
+        for gaze in dataset.gaze:
+            assert 'position' in gaze.columns
+            assert 'velocity' in gaze.columns
+            assert 'acceleration' in gaze.columns
+
+        shutil.rmtree(dataset_path, ignore_errors=True)