diff --git a/flytekit/configuration/plugin.py b/flytekit/configuration/plugin.py index 25622f4da7..05cfcd13ff 100644 --- a/flytekit/configuration/plugin.py +++ b/flytekit/configuration/plugin.py @@ -19,10 +19,15 @@ """ import os +import sys from typing import List, Optional, Protocol, runtime_checkable from click import Group -from importlib_metadata import entry_points + +if sys.version_info < (3, 10): + from importlib_metadata import entry_points +else: + from importlib.metadata import entry_points from flytekit import CachePolicy from flytekit.configuration import Config, get_config_file