Skip to content

Commit e726c09

Browse files
authored
feat: Add more __repr__ methods (feast-dev#4676)
1 parent 5291289 commit e726c09

File tree

4 files changed

+41
-4
lines changed

4 files changed

+41
-4
lines changed

sdk/python/feast/entity.py

+14
Original file line numberDiff line numberDiff line change
@@ -99,6 +99,20 @@ def __init__(
9999
self.created_timestamp = None
100100
self.last_updated_timestamp = None
101101

102+
def __repr__(self):
103+
return (
104+
f"Entity(\n"
105+
f" name={self.name!r},\n"
106+
f" value_type={self.value_type!r},\n"
107+
f" join_key={self.join_key!r},\n"
108+
f" description={self.description!r},\n"
109+
f" tags={self.tags!r},\n"
110+
f" owner={self.owner!r},\n"
111+
f" created_timestamp={self.created_timestamp!r},\n"
112+
f" last_updated_timestamp={self.last_updated_timestamp!r}\n"
113+
f")"
114+
)
115+
102116
def __hash__(self) -> int:
103117
return hash((self.name, self.join_key))
104118

sdk/python/feast/feature.py

+9-3
Original file line numberDiff line numberDiff line change
@@ -58,12 +58,18 @@ def __lt__(self, other):
5858
return self.name < other.name
5959

6060
def __repr__(self):
61-
# return string representation of the reference
62-
return f"{self.name}-{self.dtype}"
61+
return (
62+
f"Feature(\n"
63+
f" name={self._name!r},\n"
64+
f" dtype={self._dtype!r},\n"
65+
f" description={self._description!r},\n"
66+
f" labels={self._labels!r}\n"
67+
f")"
68+
)
6369

6470
def __str__(self):
6571
# readable string of the reference
66-
return f"Feature<{self.__repr__()}>"
72+
return f"Feature<{self.name}: {self.dtype}>"
6773

6874
@property
6975
def name(self):

sdk/python/feast/feature_store.py

+10
Original file line numberDiff line numberDiff line change
@@ -175,6 +175,16 @@ def version(self) -> str:
175175
"""Returns the version of the current Feast SDK/CLI."""
176176
return get_version()
177177

178+
def __repr__(self) -> str:
179+
return (
180+
f"FeatureStore(\n"
181+
f" repo_path={self.repo_path!r},\n"
182+
f" config={self.config!r},\n"
183+
f" registry={self._registry!r},\n"
184+
f" provider={self._provider!r}\n"
185+
f")"
186+
)
187+
178188
@property
179189
def registry(self) -> BaseRegistry:
180190
"""Gets the registry of this feature store."""

sdk/python/feast/field.py

+8-1
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,14 @@ def __lt__(self, other):
8181
return self.name < other.name
8282

8383
def __repr__(self):
84-
return f"Field(name='{self.name}', dtype={self.dtype}, description='{self.description}' tags={self.tags})"
84+
return (
85+
f"Field(\n"
86+
f" name={self.name!r},\n"
87+
f" dtype={self.dtype!r},\n"
88+
f" description={self.description!r},\n"
89+
f" tags={self.tags!r}\n"
90+
f")"
91+
)
8592

8693
def __str__(self):
8794
return f"Field(name={self.name}, dtype={self.dtype}, tags={self.tags})"

0 commit comments

Comments
 (0)