fix(dataset): change from_list into normal method in the alias

walledai · Aug 2, 2024 · 0a47c4e · 0a47c4e
1 parent 0b3db84
commit 0a47c4e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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):