diff --git a/walledeval/data/core.py b/walledeval/data/core.py index 1d80d66..66419d4 100644 --- a/walledeval/data/core.py +++ b/walledeval/data/core.py @@ -87,10 +87,9 @@ def from_hub(self, name, config, split, self.model, **ds_kwargs ) - @classmethod - def from_list(cls, name: str, lst: list[dict]): + def from_list(self, name: str, lst: list[dict]): return HuggingFaceDataset.from_list( - name, lst + name, lst, self.model ) def from_csv(self, filenames: Union[str, list[str]], **csv_kwargs):