fix a bug in client archieve_project method and fix lint in grpc auth

Signed-off-by: Rand Xie <[email protected]>
feast-dev · Mar 18, 2021 · 21e49b3 · 21e49b3
1 parent b7594aa
commit 21e49b3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/sdk/python/feast/client.py b/sdk/python/feast/client.py
@@ -451,7 +451,7 @@ def archive_project(self, project):
             )
         else:
             try:
-                self._core_service_stub.ArchiveProject(
+                self._core_service.ArchiveProject(
                     ArchiveProjectRequest(name=project),
                     timeout=self._config.getint(opt.GRPC_CONNECTION_TIMEOUT),
                     metadata=self._get_grpc_metadata(),

diff --git a/sdk/python/feast/grpc/__init__.py b/sdk/python/feast/grpc/__init__.py
diff --git a/sdk/python/feast/grpc/auth.py b/sdk/python/feast/grpc/auth.py
@@ -103,13 +103,13 @@ def _refresh_token(self, config: Config):
         import requests
 
         headers_token = {"content-type": "application/json"}
-        data_token = {
+        data_token_dict = {
             "grant_type": config.get(opt.OAUTH_GRANT_TYPE),
             "client_id": config.get(opt.OAUTH_CLIENT_ID),
             "client_secret": config.get(opt.OAUTH_CLIENT_SECRET),
             "audience": config.get(opt.OAUTH_AUDIENCE),
         }
-        data_token = json.dumps(data_token)
+        data_token = json.dumps(data_token_dict)
         response_token = requests.post(
             config.get(opt.OAUTH_TOKEN_REQUEST_URL),
             headers=headers_token,