fix: correct uav imports after refactoring munged and merged aidata c…

…onfig
mbari-org · Aug 28, 2024 · 9a8bf29 · 9a8bf29
1 parent 940ed36
commit 9a8bf29
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/aipipeline/projects/uav/cluster_pipeline.py b/aipipeline/projects/uav/cluster_pipeline.py
@@ -11,7 +11,7 @@
 import sys
 
 from aipipeline.docker.utils import run_docker
-from aipipeline.projects.uav_901902.args_common import parse_args, POSSIBLE_PLATFORMS
+from aipipeline.projects.uav.args_common import parse_args, POSSIBLE_PLATFORMS
 from config_setup import setup_config
 
 logger = logging.getLogger(__name__)

diff --git a/aipipeline/projects/uav/config/config.yml b/aipipeline/projects/uav/config/config.yml
@@ -16,6 +16,20 @@ docker:
     /tmp:
       bind: /tmp
       mode: rw
+    /mnt/UAV:
+      bind: /mnt/UAV
+      mode: rw
+
+mounts:
+  - name: "image"
+    path: "/mnt/UAV"
+    host: "mantis.shore.mbari.org"
+    nginx_root: "/UAV"
+
+plugins:
+  - name: "extractor"
+    module: "aidata.plugins.extractors.tap_sony_media"
+    function: "extract_media"
 
 data:
   raw_path: "/mnt/UAV/"

diff --git a/aipipeline/projects/uav/detect_pipeline.py b/aipipeline/projects/uav/detect_pipeline.py
@@ -11,7 +11,7 @@
 import logging
 
 from aipipeline.docker.utils import run_docker
-from aipipeline.projects.uav_901902.args_common import parse_args, POSSIBLE_PLATFORMS
+from aipipeline.projects.uav.args_common import parse_args, POSSIBLE_PLATFORMS
 from config_setup import setup_config
 
 logger = logging.getLogger(__name__)