Fix issues with lint test and upgrade pip version (#1964)

.github/workflows/integration_tests.yml

@@ -50,6 +50,9 @@ jobs:
           aws-region: us-west-2
       - name: Use AWS CLI
         run: aws sts get-caller-identity
+      - name: Upgrade pip version
+        run: |
+          pip install --upgrade "pip>=21.3.1"
       - name: Get pip cache dir
         id: pip-cache
         run: |

.github/workflows/linter.yml

@@ -13,6 +13,9 @@ jobs:
         with:
           python-version: "3.7"
           architecture: x64
+      - name: Upgrade pip version
+        run: |
+          pip install --upgrade "pip>=21.3.1"
       - name: Get pip cache dir
         id: pip-cache
         run: |

.github/workflows/pr_integration_tests.yml

@@ -61,6 +61,9 @@ jobs:
           aws-region: us-west-2
       - name: Use AWS CLI
         run: aws sts get-caller-identity
+      - name: Upgrade pip version
+        run: |
+          pip install --upgrade "pip>=21.3.1"
       - name: Get pip cache dir
         id: pip-cache
         run: |

.github/workflows/unit_tests.yml

@@ -20,9 +20,9 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
           architecture: x64
-      - name: Downgrade  pip
+      - name: Upgrade pip version
         run: |
-          pip install --upgrade "pip<21.3"
+          pip install --upgrade "pip>=21.3.1"
       - name: Get pip cache dir
         id: pip-cache
         run: |

sdk/python/feast/infra/gcp.py

@@ -22,7 +22,7 @@ class GCSRegistryStore(RegistryStore):
     def __init__(self, registry_config: RegistryConfig, repo_path: Path):
         uri = registry_config.path
         try:
-            from google.cloud import storage
+            import google.cloud.storage as storage
         except ImportError as e:
             from feast.errors import FeastExtrasDependencyImportError
 
@@ -34,7 +34,7 @@ def __init__(self, registry_config: RegistryConfig, repo_path: Path):
         self._blob = self._uri.path.lstrip("/")
 
     def get_registry_proto(self):
-        from google.cloud import storage
+        import google.cloud.storage as storage
         from google.cloud.exceptions import NotFound
 
         file_obj = TemporaryFile()