Skip to content

Commit

Permalink
add validation and unittest for dataset materalize col_stats
Browse files Browse the repository at this point in the history
  • Loading branch information
HoustonJ2013 committed Dec 27, 2024
1 parent 7a44362 commit b089036
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch_frame/data/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -614,7 +614,7 @@ def materialize(
index, value = ser.index.tolist(), ser.values.tolist()
self._col_stats[col][StatType.COUNT] = (index, value)
else:
# basic validation for the for col_stats provided by the user
# basic validation for the col_stats provided by the user
for col_, stype_ in self.col_to_stype.items():
assert col_ in col_stats, \
f"{col_} is not specified in the provided col_stats"
Expand Down

0 comments on commit b089036

Please sign in to comment.