chore(deps): Run task alone without predictor

hotosm · Jan 9, 2025 · 5533497 · 5533497
1 parent 6663cd7
commit 5533497
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/backend/api-requirements.txt b/backend/api-requirements.txt
@@ -19,8 +19,6 @@ gpxpy==1.5.0
 geojson2osm==0.0.1
 osmconflator==0.0.11
 orthogonalizer==0.0.4
-fairpredictor==0.0.37
-opencv-python-headless==4.10.0.84
 rasterio==1.3.8
 numpy<2.0.0
 mercantile==1.2.1 

diff --git a/backend/core/tasks.py b/backend/core/tasks.py
@@ -33,7 +33,6 @@
 from django.contrib.gis.geos import GEOSGeometry
 from django.shortcuts import get_object_or_404
 from django.utils import timezone
-from predictor import download_imagery, get_start_end_download_coords
 
 from .utils import S3Uploader
 
@@ -115,6 +114,8 @@ def get_file_count(path):
 
 
 def prepare_data(training_instance, dataset_id, feedback, zoom_level, source_imagery):
+    from predictor import download_imagery, get_start_end_download_coords
+
     training_input_base_path = os.path.join(
         settings.TRAINING_WORKSPACE, f"dataset_{dataset_id}"
     )

diff --git a/backend/requirements.txt b/backend/requirements.txt
@@ -2,4 +2,6 @@
 hot-fair-utilities==2.0.8
 tflite-runtime==2.14.0
 tippecanoe==2.45.0
+fairpredictor==0.0.37
+opencv-python-headless==4.10.0.84
 onnxruntime>=1.0.0