GitHub CI fixes (#161)

Summary: Pull Request resolved: #161 Differential Revision: D50707779
pytorch · Oct 26, 2023 · 8083e83 · 8083e83
1 parent cfc299b
commit 8083e83
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -1,5 +1,6 @@
 aiobotocore
 boto3
+expecttest
 google-cloud-storage
 google-resumable-media
 numpy

diff --git a/tests/test_manifest.py b/tests/test_manifest.py
@@ -26,7 +26,8 @@
 )
 from torchsnapshot.manifest_ops import get_manifest_for_rank
 from torchsnapshot.manifest_utils import is_fully_replicated_entry
-from torchsnapshot.tests.assets.manifest import (
+
+from .assets.manifest import (
     _MANIFEST_0,
     _MANIFEST_1,
     _MANIFEST_2,