Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
jafermarq committed Dec 16, 2024
1 parent ac3bc68 commit eaf9ac3
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -182,7 +182,7 @@ def load_partition(self, partition_id: int) -> datasets.Dataset:
assert self._partition_columns is not None
if partition_id < 0 or partition_id >= len(self._partition_columns):
raise IndexError(
f"partition_id: {partition_id} out of range <0, {len(self.num_partitions) - 1}>."
f"partition_id: {partition_id} out of range <0, {self.num_partitions - 1}>."
)
columns = self._partition_columns[partition_id]
return self.dataset.select_columns(columns)
Expand Down

0 comments on commit eaf9ac3

Please sign in to comment.