check azure uri has the right account name

Signed-off-by: Jacob Klegar <[email protected]>
feast-dev · Dec 28, 2020 · 1a9a10a · 1a9a10a
1 parent 2c4f0fc
commit 1a9a10a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/python/feast/staging/storage_client.py b/sdk/python/feast/staging/storage_client.py
@@ -374,7 +374,7 @@ def list_files(self, uri: ParseResult) -> List[str]:
             return [f"{self.account_url}/{bucket}/{path}"]
 
     def _uri_to_bucket_key(self, uri: ParseResult) -> Tuple[str, str]:
-        assert uri.hostname is not None
+        assert uri.hostname == urlparse(self.account_url).hostname
         bucket = uri.path.lstrip("/").split("/")[0]
         key = uri.path.lstrip("/").split("/", 1)[1]
         return bucket, key