remove registry fixes

feathr-ai · Nov 10, 2022 · 78695d1 · 78695d1
1 parent 858f362
commit 78695d1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 42 deletions.
diff --git a/feathr_project/feathr/registry/_feathr_registry_client_aws.py b/feathr_project/feathr/registry/_feathr_registry_client_aws.py
diff --git a/feathr_project/feathr/registry/feature_registry.py b/feathr_project/feathr/registry/feature_registry.py
@@ -55,14 +55,10 @@ def save_to_feature_config_from_context(self, anchor_list, derived_feature_list,
 def default_registry_client(project_name: str, config_path:str = "./feathr_config.yaml", project_registry_tag: Dict[str, str]=None, credential = None) -> FeathrRegistry:
     from feathr.registry._feathr_registry_client import _FeatureRegistry
     from feathr.registry._feature_registry_purview import _PurviewRegistry
-    from feathr.registry._feathr_registry_client_aws import _FeatureRegistryAWS
-    from aws_requests_auth.aws_auth import AWSRequestsAuth
-
+
     envutils = _EnvVaraibleUtil(config_path)
     registry_endpoint = envutils.get_environment_variable_with_default("feature_registry", "api_endpoint")
-    if registry_endpoint and isinstance(credential, AWSRequestsAuth):
-        return _FeatureRegistryAWS(project_name, endpoint=registry_endpoint, project_tags=project_registry_tag, credential=credential)
-    elif registry_endpoint:
+    if registry_endpoint:
         return _FeatureRegistry(project_name, endpoint=registry_endpoint, project_tags=project_registry_tag, credential=credential)
     else:
         registry_delimiter = envutils.get_environment_variable_with_default('feature_registry', 'purview', 'delimiter')

diff --git a/feathr_project/setup.py b/feathr_project/setup.py
@@ -84,8 +84,7 @@
         # it brings a different version of msrest(0.7.0) which is incompatible with azure-core==1.22.1. Hence we need to pin it.
         # See this for more details: https://github.com/Azure/azure-sdk-for-python/issues/24765
         "msrest<=0.6.21",
-        "typing_extensions>=4.2.0",
-        "aws-requests-auth>=0.4.3"
+        "typing_extensions>=4.2.0"
     ],
     tests_require=[  # TODO: This has been depricated
         "pytest",