fix-unit test

aimclub · Feb 14, 2024 · a136c0f · a136c0f
1 parent bc0c56c
commit a136c0f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/unit-test.yaml b/.github/workflows/unit-test.yaml
@@ -22,9 +22,9 @@ jobs:
     - name: Download dataset
       run: |
         cd user_datasets
-        mkdir aeral
-        cd aeral
-        curl -L "https://public.roboflow.com/ds/et6MNsOwMJ?key=GMSDIzpYnW" > roboflow.zip; unzip roboflow.zip; rm roboflow.zip
+        git clone https://github.com/AIRI-Institute/WaRP
+        mkdir WaRP/Warp-D/valid
+        cp -r WaRP/Warp-D/test/* WaRP/Warp-D/valid/
     - name: Test
       run: |
         python tests/unit_test.py 
diff --git a/ODRS/train_utils/config/custom_config.yaml b/ODRS/train_utils/config/custom_config.yaml
@@ -1,7 +1,7 @@
 BATCH_SIZE: 20
 CLASSES: classes.txt
 DATA_PATH: /home/runner/work/ODRS/ODRS/user_datasets/WaRP/Warp-D
-EPOCHS: 2
+EPOCHS: 1
 GPU_COUNT: 1
 IMG_SIZE: 300
 MODEL: yolov8s

diff --git a/tests/unit_test.py b/tests/unit_test.py
@@ -8,7 +8,7 @@
 
 class TestDatasetInfo(unittest.TestCase):
     def test_dataset_info(self):
-        dataset_path = "/home/runner/work/ODRS/ODRS/user_datasets/yolo/aeral"
+        dataset_path = "/home/runner/work/ODRS/ODRS/user_datasets/WaRP/Warp-D"
         classes_path = "/home/runner/work/ODRS/ODRS/classes.txt"
 
         result = dataset_info(dataset_path, classes_path)
@@ -22,7 +22,7 @@ def test_dataset_info(self):
         self.assertIsInstance(result[4], float)
 
     def test_process_directory_img(self):
-        test_dir = "/home/runner/work/ODRS/ODRS/user_datasets/yolo/aeral/test"
+        test_dir = "/home/runner/work/ODRS/ODRS/user_datasets/yolo/WaRP/Warp-D/test"
         os.makedirs(test_dir, exist_ok=True)
 
         label_path = os.path.join(test_dir, 'labels')