fix test

Signed-off-by: Danny Chiao <[email protected]>
feast-dev · Feb 14, 2022 · 3feecfa · 3feecfa
1 parent c002451
commit 3feecfa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/sdk/python/feast/infra/online_stores/datastore.py b/sdk/python/feast/infra/online_stores/datastore.py
@@ -336,7 +336,6 @@ class DatastoreTable(InfraObject):
     """
 
     project: str
-    name: str
     project_id: Optional[str]
     namespace: Optional[str]
 
@@ -347,8 +346,8 @@ def __init__(
         project_id: Optional[str] = None,
         namespace: Optional[str] = None,
     ):
+        super().__init__(name)
         self.project = project
-        self.name = name
         self.project_id = project_id
         self.namespace = namespace
 

diff --git a/sdk/python/feast/infra/online_stores/dynamodb.py b/sdk/python/feast/infra/online_stores/dynamodb.py
@@ -227,11 +227,10 @@ class DynamoDBTable(InfraObject):
         region: The region of the table.
     """
 
-    name: str
     region: str
 
     def __init__(self, name: str, region: str):
-        self.name = name
+        super().__init__(name)
         self.region = region
 
     def to_infra_object_proto(self) -> InfraObjectProto: