Data Ingestion Modified

utpalpaul108 · Nov 27, 2023 · a0c9313 · a0c9313
1 parent 5f4bb35
commit a0c9313
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/notebooks/data_ingestion.ipynb b/notebooks/data_ingestion.ipynb
@@ -152,8 +152,8 @@
     "        Preprocess the raw dataset\n",
     "        '''\n",
     "        if os.path.exists(self.config.raw_dataset_dir):\n",
-    "            create_directories([self.config.dataset_dir])\n",
     "            try:\n",
+    "                create_directories([self.config.dataset_dir])\n",
     "                for root, dirs, files in os.walk(self.config.raw_dataset_dir):\n",
     "                    for file in files:\n",
     "                        \n",

diff --git a/src/wasteDetection/components/data_ingestion.py b/src/wasteDetection/components/data_ingestion.py
@@ -39,8 +39,8 @@ def _preprocess_dataset(self):
         Preprocess the raw dataset
         '''
         if os.path.exists(self.config.raw_dataset_dir):
-            create_directories([self.config.dataset_dir])
             try:
+                create_directories([self.config.dataset_dir])
                 for root, dirs, files in os.walk(self.config.raw_dataset_dir):
                     for file in files: