method gets multiple values

aeye-lab · Sep 14, 2023 · be90ae4 · be90ae4
1 parent 6ea6b77
commit be90ae4
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/pymovements/dataset/dataset.py b/src/pymovements/dataset/dataset.py
@@ -36,7 +36,6 @@
 from pymovements.events.frame import EventDataFrame
 from pymovements.events.processing import EventGazeProcessor
 from pymovements.gaze import GazeDataFrame
-from pymovements.gaze.transforms import TransformLibrary
 
 
 class Dataset:
@@ -245,11 +244,8 @@ def apply(
         self._check_gaze_dataframe()
 
         disable_progressbar = not verbose
-        for gaze_df in tqdm(self.gaze, disable=disable_progressbar):
-            if method in TransformLibrary:
-                gaze_df.transform(method, **kwargs)
-            else:
-                raise ValueError(method)
+        for gaze in tqdm(self.gaze, disable=disable_progressbar):
+            gaze.apply(method, **kwargs)
 
         return self