fix test

adchia · adchia · commit 3feecfa2f0d9 · 2022-02-13T20:15:01.000-05:00
Signed-off-by: Danny Chiao &lt;danny@tecton.ai&gt;
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:
