removing base_module] fbcode/vision/fair/pytorchvideo/pytorchvideo_tr…

…ainer/tests/TARGETS Reviewed By: azad-meta Differential Revision: D61014797 fbshipit-source-id: cf686d20c5bc73c70e4ba0e6542b87e9e8a1a5fc
sanchitv7 · Aug 13, 2024 · e02c17b · e02c17b
1 parent 1fadaef
commit e02c17b
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 5 deletions.
diff --git a/pytorchvideo_trainer/tests/test_task_byol.py b/pytorchvideo_trainer/tests/test_task_byol.py
@@ -2,7 +2,7 @@
 
 # pyre-strict
 from torchrecipes.core.base_train_app import BaseTrainApp
-from util import (
+from vision.fair.pytorchvideo.pytorchvideo_trainer.tests.util import (
     BaseTrainAppTestCase,
     create_small_kinetics_dataset,
     run_locally,

diff --git a/pytorchvideo_trainer/tests/test_task_moco_v2.py b/pytorchvideo_trainer/tests/test_task_moco_v2.py
@@ -2,7 +2,7 @@
 
 # pyre-strict
 from torchrecipes.core.base_train_app import BaseTrainApp
-from util import (
+from vision.fair.pytorchvideo.pytorchvideo_trainer.tests.util import (
     BaseTrainAppTestCase,
     create_small_kinetics_dataset,
     run_locally,

diff --git a/pytorchvideo_trainer/tests/test_task_module_all.py b/pytorchvideo_trainer/tests/test_task_module_all.py
@@ -10,7 +10,11 @@
 from pytorch_lightning import Trainer
 from pytorchvideo_trainer.datamodule.datamodule import VideoClassificationDataModuleConf
 from pytorchvideo_trainer.train_app import VideoClassificationTrainAppConf
-from util import create_small_kinetics_dataset, run_locally, tempdir
+from vision.fair.pytorchvideo.pytorchvideo_trainer.tests.util import (
+    create_small_kinetics_dataset,
+    run_locally,
+    tempdir,
+)
 
 
 class TestMain(unittest.TestCase):

diff --git a/pytorchvideo_trainer/tests/test_task_simclr.py b/pytorchvideo_trainer/tests/test_task_simclr.py
@@ -2,7 +2,7 @@
 
 # pyre-strict
 from torchrecipes.core.base_train_app import BaseTrainApp
-from util import (
+from vision.fair.pytorchvideo.pytorchvideo_trainer.tests.util import (
     BaseTrainAppTestCase,
     create_small_kinetics_dataset,
     run_locally,

diff --git a/pytorchvideo_trainer/tests/test_task_video_classification.py b/pytorchvideo_trainer/tests/test_task_video_classification.py
@@ -2,7 +2,7 @@
 
 # pyre-strict
 from torchrecipes.core.base_train_app import BaseTrainApp
-from util import (
+from vision.fair.pytorchvideo.pytorchvideo_trainer.tests.util import (
     BaseTrainAppTestCase,
     create_small_kinetics_dataset,
     run_locally,