aws default region name configurable

Signed-off-by: lblokhin <lenin133@yandex.ru>
feast-dev · feast-ci-bot · Jul 3, 2021 · Apr 27, 2021 · Apr 27, 2021 · May 4, 2021
commit 6a99cd9e07a98559f7db2d5cddaf64994c480cb4
diff --git a/sdk/python/feast/infra/aws_dynamodb_provider.py b/sdk/python/feast/infra/aws_dynamodb_provider.py
@@ -1,3 +1,4 @@
+import os
 from datetime import datetime
 from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union
 
@@ -41,8 +42,15 @@ def __init__(self, config: RepoConfig):
         else:
             self._wcu = 5
 
+        if config and config.online_store and config.online_store.region_name:
+            self.region_name = config.online_store.region_name
+            os.environ["AWS_DEFAULT_REGION"] = self.region_name
+        else:
+            self.region_name = os.environ.get("AWS_DEFAULT_REGION", "us-west-2")
+            os.environ["AWS_DEFAULT_REGION"] = self.region_name
+
     def _initialize_dynamodb(self):
-        return boto3.resource("dynamodb")
+        return boto3.resource("dynamodb", region_name=self.region_name)
 
     def update_infra(
         self,

@@ -53,6 +53,9 @@ class DynamoDbOnlineStoreConfig(FeastBaseModel):
     wcu: Optional[PositiveInt] = 5
     """ Write capacity unit """
 
+    region_name: Optional[StrictStr] = None
+    """ AWS Region Name """
+
 
 OnlineStoreConfig = Union[
     DatastoreOnlineStoreConfig, SqliteOnlineStoreConfig, DynamoDbOnlineStoreConfig