Fix

sakentsunofu · Jan 10, 2024 · ab3b1bc · ab3b1bc
1 parent 9fabfb2
commit ab3b1bc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/dataset_preparer.py b/src/dataset_preparer.py
@@ -22,6 +22,7 @@ def __init__(self, dataset_name, files):
         self.dataset_name = dataset_name
         self.files = files
         self.dataset_dir = f"data/{self.dataset_name}/"
+        self.filename = self.files[0].split('/')[-1].split('.')[0]
         logging.basicConfig(level=logging.INFO)
 
     def download(self):