cleanup

Author: chenmoneygithub

tests/datasets/test_dataset.py

@@ -16,8 +16,8 @@
 
 
 class CSVDataset(Dataset):
-    def __init__(self, file_path, input_keys=None, *args, **kwargs) -> None:
-        super().__init__(input_keys=input_keys, *args, **kwargs)
+    def __init__(self, file_path, input_keys=None, **kwargs) -> None:
+        super().__init__(input_keys=input_keys, **kwargs)
         df = pd.read_csv(file_path)
         data = df.to_dict(orient="records")
         self._train = [