Skip to content

Commit f1fb525

Browse files
authored
Merge pull request #326 from appliedAI-Initiative/298-iteration-in-dataset-is-broken
Remove bogus __iter__ method.
2 parents aa8aa46 + a0bae52 commit f1fb525

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

src/pydvl/utils/dataset.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -130,9 +130,6 @@ def make_names(s: str, a: np.ndarray) -> List[str]:
130130
self._indices = np.arange(len(self.x_train))
131131
self._data_names = data_names if data_names is not None else self._indices
132132

133-
def __iter__(self):
134-
return self.x_train, self.y_train, self.x_test, self.y_test
135-
136133
def __getitem__(self, idx: Union[int, slice, Iterable]) -> Tuple:
137134
return self.x_train[idx], self.y_train[idx]
138135

0 commit comments

Comments
 (0)