diff --git a/sdk/python/feast/permissions/server/utils.py b/sdk/python/feast/permissions/server/utils.py index ac70f187ce..9a8b319dbc 100644 --- a/sdk/python/feast/permissions/server/utils.py +++ b/sdk/python/feast/permissions/server/utils.py @@ -11,7 +11,6 @@ AuthManager, set_auth_manager, ) -from feast.permissions.auth.kubernetes_token_parser import KubernetesTokenParser from feast.permissions.auth.oidc_token_parser import OidcTokenParser from feast.permissions.auth.token_extractor import TokenExtractor from feast.permissions.auth.token_parser import TokenParser @@ -116,6 +115,10 @@ def init_auth_manager( raise ValueError(f"Unmanaged server type {server_type}") if auth_type == AuthManagerType.KUBERNETES: + from feast.permissions.auth.kubernetes_token_parser import ( + KubernetesTokenParser, + ) + token_parser = KubernetesTokenParser() elif auth_type == AuthManagerType.OIDC: assert isinstance(auth_config, OidcAuthConfig) diff --git a/setup.py b/setup.py index 6da5e8226a..f763fb3f1d 100644 --- a/setup.py +++ b/setup.py @@ -63,7 +63,6 @@ "psutil", "bigtree>=0.19.2", "pyjwt", - "kubernetes<=20.13.0", ] GCP_REQUIRED = [