wip: fixed bugs

Dockerfiles/gpu-310.Dockerfile

@@ -92,7 +92,7 @@ COPY requirements.txt extra-requirements.txt setup.py /tmp/
 RUN python3.10 -m venv /opt/venv
 ENV PATH="/opt/venv/bin:${PATH}"
 RUN python3 -m pip install --no-cache-dir -U pip==22.0.4 setuptools==53.0.0 wheel==0.36.2
-RUN python3 -m pip install --no-cache-dir install --upgrade setuptools
+RUN python3 -m pip install --no-cache-dir --upgrade setuptools
 RUN python3 -m pip install "pybind11[global]" # This prevents "ModuleNotFoundError: No module named 'pybind11'"
 
 RUN python3 -m pip install intel-openmp

README-GB.md

@@ -186,7 +186,7 @@ Overwrite the container `ENTRYPOINT` by using `--entrypoint` from command line a
 `nvidia-smi`
 
 ```sh
-docker run -it --rm  --gpus all --entrypoint /bin/bash marieai/marie:3.0.22-cuda
+docker run -it --rm  --gpus all --entrypoint /bin/bash marieai/marie:3.0.31-cuda
 ```
 
 Remove dangling containers

README.md

@@ -134,6 +134,7 @@ aws s3 rm  s3://marie --recursive --profile marie --endpoint-url http://localhos
 
 
 
+
 # Production setup
 
 

extra-requirements.txt

@@ -168,3 +168,4 @@ seaborn:                      standard,core
 textual:                      standard,core
 textual_textarea:             standard,core
 textual_fastdatatable:        standard,core
+etcd3:                        standard,core

marie/pipe/extract_pipeline.py

@@ -318,6 +318,7 @@ def execute_frames_pipeline(
             "ref_id": ref_id,
             "ref_type": ref_type,
             "job_id": job_id,
+            "work_dir": root_asset_dir,
             "pipeline": self.pipeline_name,
             "pages": f"{len(frames)}",
         }
@@ -437,6 +438,7 @@ def execute_regions_pipeline(
             "ref_id": ref_id,
             "ref_type": ref_type,
             "job_id": job_id,
+            "work_dir": root_asset_dir,
             "pages": f"{len(frames)}",
             "ocr": results,
         }

marie/resources/extra-requirements.txt

@@ -167,4 +167,5 @@ efficientnet-pytorch:         standard,core
 seaborn:                      standard,core
 textual:                      standard,core
 textual_textarea:             standard,core
-textual_fastdatatable:        standard,core
+textual_fastdatatable:        standard,core
+etcd3:                        standard,core

marie/serve/runtimes/worker/batch_queue.py

@@ -3,7 +3,7 @@
 from asyncio import Event, Task
 from typing import TYPE_CHECKING, Callable, Dict, List, Optional, Union
 
-from jina._docarray import docarray_v2
+from marie._docarray import docarray_v2
 
 if not docarray_v2:
     from docarray import DocumentArray