Fixes

Lightning-Universe · Aug 13, 2021 · 3b03eba · 3b03eba
1 parent f71f5e3
commit 3b03eba
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/flash/video/classification/data.py b/flash/video/classification/data.py
@@ -75,6 +75,9 @@ def load_data(self, data: str, dataset: Optional[Any] = None) -> "LabeledVideoDa
             dataset.num_classes = len(np.unique([s[1]["label"] for s in ds._labeled_videos]))
         return ds
 
+    def load_sample(self, sample):
+        return sample
+
     def predict_load_sample(self, sample: Dict[str, Any]) -> Dict[str, Any]:
         video_path = sample[DefaultDataKeys.INPUT]
         sample.update(self._encoded_video_to_dict(EncodedVideo.from_path(video_path)))