diff --git a/sdk/python/feast/infra/aws_dynamo_provider.py b/sdk/python/feast/infra/aws_dynamo_provider.py index 57444404bd..53c84f69af 100644 --- a/sdk/python/feast/infra/aws_dynamo_provider.py +++ b/sdk/python/feast/infra/aws_dynamo_provider.py @@ -24,14 +24,9 @@ class AwsDynamoProvider(Provider): - _aws_project_id: Optional[str] def __init__(self, config: RepoConfig): assert isinstance(config.online_store, DynamoOnlineStoreConfig) - if config and config.online_store and config.online_store.project_id: - self._aws_project_id = config.online_store.project_id - else: - self._aws_project_id = None def _initialize_dynamodb(self): return boto3.resource('dynamodb') diff --git a/sdk/python/feast/registry.py b/sdk/python/feast/registry.py index d2f4ef967a..9782471588 100644 --- a/sdk/python/feast/registry.py +++ b/sdk/python/feast/registry.py @@ -540,10 +540,13 @@ def _write_registry(self, registry_proto: RegistryProto): registry_proto.version_id = str(uuid.uuid4()) registry_proto.last_updated.FromDatetime(datetime.utcnow()) # we have already checked the bucket exists so no need to do it again - registry_bucket = boto3.resource('s3').Bucket(self._bucket) - registry_db = registry_bucket.Object(self._key) file_obj = TemporaryFile() file_obj.write(registry_proto.SerializeToString()) file_obj.seek(0) - registry_db.upload_fileobj(file_obj) - return + s3 = boto3.client('s3') + s3.put_object( + Bucket=self._bucket, + Body=file_obj, + Key=self._key + ) + return \ No newline at end of file diff --git a/sdk/python/feast/repo_config.py b/sdk/python/feast/repo_config.py index 05b5557e45..6f5ac00612 100644 --- a/sdk/python/feast/repo_config.py +++ b/sdk/python/feast/repo_config.py @@ -38,7 +38,6 @@ class DynamoOnlineStoreConfig(FeastBaseModel): """Online store config for DynamoDB store""" type: Literal["dynamo"] = "dynamo" """Online store type selector""" - project_id: Optional[StrictStr] = None OnlineStoreConfig = Union[DatastoreOnlineStoreConfig, SqliteOnlineStoreConfig, DynamoOnlineStoreConfig]