fix: pre commit issues

Author: Saumya Saksena

.github/workflows/ci.yml

@@ -2,9 +2,9 @@ name: CI
 
 on:
   push:
-    branches: [main, develop]
+    branches: [main, dev, master]
   pull_request:
-    branches: [main, develop]
+    branches: [main, dev, master]
 
 jobs:
   test:
@@ -13,7 +13,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-latest, windows-latest]
-        python-version: ["3.8", "3.9", "3.10", "3.11"]
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
 
     steps:
       - uses: actions/checkout@v3
@@ -109,9 +109,9 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install safety bandit
+          pip install pre-commit
+          pre-commit install
 
-      - name: Check for security vulnerabilities
+      - name: Check for pre-commit issues
         run: |
-          safety check --json || true
-          bandit -r src/ -f json || true
+          pre-commit run --all-files

setup.py

@@ -18,7 +18,8 @@
     version="2.0.0",
     author="dronefreak",
     author_email="",
-    description="Real-time monocular depth estimation for DJI Tello drone collision avoidance using PyDNet",
+    description="Real-time monocular depth estimation for DJI "
+    "Tello drone collision avoidance using PyDNet",
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/dronefreak/dji-tello-collision-avoidance-pydnet",
@@ -59,7 +60,8 @@
     },
     include_package_data=True,
     zip_safe=False,
-    keywords="drone tello depth-estimation computer-vision pytorch collision-avoidance autonomous-navigation",
+    keywords="drone tello depth-estimation computer-vision pytorch"
+    "collision-avoidance autonomous-navigation",
     project_urls={
         "Bug Reports": "https://github.com/dronefreak/dji-tello-collision-avoidance-pydnet/issues",
         "Source": "https://github.com/dronefreak/dji-tello-collision-avoidance-pydnet",

src/collision_avoidance.py

@@ -71,9 +71,7 @@ def analyze_depth(self, depth_map: np.ndarray) -> Dict[str, Any]:
         center_y, center_x = h // 2, w // 2
         h4 = max(h // 4, 1)
         w4 = max(w // 4, 1)
-        center_region = depth_map[
-            center_y - h4 : center_y + h4, center_x - w4 : center_x + w4
-        ]
+        center_region = depth_map[center_y - h4 : center_y + h4, center_x - w4 : center_x + w4]
         if center_region.size == 0:
             center_region = depth_map
         center_depth = float(np.mean(center_region))

tests/test_collision_avoidance.py

@@ -12,7 +12,7 @@
 
 # Check if OpenCV is available (required for collision_avoidance)
 try:
-    import cv2
+    import cv2  # noqa: F401
 
     CV2_AVAILABLE = True
 except ImportError:

tests/test_depth_estimator.py

@@ -13,7 +13,7 @@
 
 # Check if TensorFlow is available
 try:
-    import tensorflow as tf
+    import tensorflow as tf  # noqa: F401
 
     TF_AVAILABLE = True
     from src.depth_estimator import DepthEstimator, PyDNetModel

tests/test_tello_interface.py

@@ -15,7 +15,7 @@
 
 # Check if OpenCV is available (required for imports through src package)
 try:
-    import cv2
+    import cv2  # noqa: F401
 
     CV2_AVAILABLE = True
 except ImportError:

tests/test_utils.py

@@ -11,12 +11,20 @@
 
 # Check if OpenCV is available (required for utils module)
 try:
-    import cv2
+    import cv2  # noqa: F401
 
     CV2_AVAILABLE = True
-    from src.utils import (apply_colormap, draw_center_region, draw_crosshair, draw_depth_stats,
-                           draw_fps, find_max_depth_region, preprocess_image, resize_with_aspect_ratio,
-                           visualize_depth)
+    from src.utils import (
+        apply_colormap,
+        draw_center_region,
+        draw_crosshair,
+        draw_depth_stats,
+        draw_fps,
+        find_max_depth_region,
+        preprocess_image,
+        resize_with_aspect_ratio,
+        visualize_depth,
+    )
 except ImportError:
     CV2_AVAILABLE = False
 

tests/test_webcam_demo.py

@@ -15,7 +15,7 @@
 
 # Check if OpenCV is available (required for webcam_source)
 try:
-    import cv2
+    import cv2  # noqa: F401
 
     CV2_AVAILABLE = True
 except ImportError: