Move sys.path insert earlier in s3op.py (#1098)

Netflix · Aug 3, 2022 · f11a372 · f11a372
1 parent 925530b
commit f11a372
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/metaflow/datatools/s3op.py b/metaflow/datatools/s3op.py
@@ -25,6 +25,12 @@
     from urllib.parse import urlparse
     from queue import Full as QueueFull
 
+if __name__ == "__main__":
+    # When launched standalone, point to our parent metaflow
+    sys.path.insert(
+        0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../../"))
+    )
+
 from metaflow._vendor import click
 
 # we use Metaflow's parallel_imap_unordered instead of
@@ -1052,9 +1058,4 @@ def info(
 
 
 if __name__ == "__main__":
-    # When launched standalone, point to our parent metaflow
-    sys.path.insert(
-        0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../../"))
-    )
-
     cli(auto_envvar_prefix="S3OP")