fix imports in stubs and extensions_support

Netflix · May 15, 2024 · 2f87ca0 · 2f87ca0
1 parent d7b1d54
commit 2f87ca0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/metaflow/cmd/develop/stubs.py b/metaflow/cmd/develop/stubs.py
@@ -24,7 +24,7 @@ def _check_stubs_supported():
             if _py_ver >= (3, 8):
                 from importlib import metadata
             elif _py_ver >= (3, 6):
-                from metaflow._vendor.v3_6 import importlib_metadata as metadata
+                from metaflow._vendor import importlib_metadata as metadata
             else:
                 from metaflow._vendor.v3_5 import importlib_metadata as metadata
             _metadata_package = metadata

diff --git a/metaflow/extension_support/__init__.py b/metaflow/extension_support/__init__.py
@@ -263,7 +263,7 @@ def multiload_all(modules, extension_point, dst_globals):
     if _py_ver >= (3, 8):
         from importlib import metadata
     elif _py_ver >= (3, 6):
-        from metaflow._vendor.v3_6 import importlib_metadata as metadata
+        from metaflow._vendor import importlib_metadata as metadata
     else:
         from metaflow._vendor.v3_5 import importlib_metadata as metadata
     _mfext_supported = True