arkindex_worker/image.py

@@ -60,7 +60,7 @@ def open_image(path, mode="RGB", rotation_angle=0, mirrored=False):
         image = image.convert(mode)
 
     if mirrored:
-        image = image.transpose(Image.FLIP_LEFT_RIGHT)
+        image = image.transpose(Image.Transpose.FLIP_LEFT_RIGHT)
 
     if rotation_angle:
         image = image.rotate(-rotation_angle, expand=True)
@@ -102,7 +102,9 @@ def download_image(url):
     # Preprocess the image and prepare it for classification
     image = Image.open(BytesIO(resp.content))
     logger.info(
-        "Downloaded image {} - size={}x{}".format(url, image.size[0], image.size[1])
+        "Downloaded image {} - size={}x{} in {}".format(
+            url, image.size[0], image.size[1], resp.elapsed
+        )
     )
 
     return image

tests/test_base_worker.py

@@ -130,7 +130,7 @@ def test_cli_envvar_debug_given(mocker, monkeypatch, mock_worker_run_api):
 
     assert logger.level == logging.NOTSET
     mocker.patch.object(sys, "argv", ["worker"])
-    monkeypatch.setenv("ARKINDEX_DEBUG", True)
+    monkeypatch.setenv("ARKINDEX_DEBUG", "True")
     worker.args = worker.parser.parse_args()
     assert worker.is_read_only is False
     assert worker.worker_version_id == "12341234-1234-1234-1234-123412341234"

worker-{{cookiecutter.slug}}/.pre-commit-config.yaml

 repos:
-  - repo: https://github.com/asottile/seed-isort-config
-    rev: v2.2.0
-    hooks:
-      - id: seed-isort-config
   - repo: https://github.com/pre-commit/mirrors-isort
-    rev: v4.3.21
+    rev: v5.10.1
     hooks:
       - id: isort
   - repo: https://github.com/ambv/black
-    rev: 22.3.0
+    rev: 22.6.0
     hooks:
     - id: black
   - repo: https://gitlab.com/pycqa/flake8
-    rev: 3.8.3
+    rev: 3.9.2
     hooks:
       - id: flake8
         additional_dependencies:
@@ -20,7 +16,7 @@ repos:
           - 'flake8-copyright==0.2.2'
           - 'flake8-debugger==3.1.0'
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v3.1.0
+    rev: v4.3.0
     hooks:
       - id: check-ast
       - id: check-docstring-first
@@ -37,7 +33,7 @@ repos:
       - id: check-json
       - id: requirements-txt-fixer
   - repo: https://github.com/codespell-project/codespell
-    rev: v1.17.1
+    rev: v2.2.1
     hooks:
       - id: codespell
         args: ['--write-changes']

worker-{{cookiecutter.slug}}/tests/conftest.py

@@ -16,5 +16,9 @@ def setup_environment(responses):
     )
     responses.add_passthru(schema_url)
 
+    # Set schema url in environment
+    os.environ["ARKINDEX_API_SCHEMA_URL"] = schema_url
     # Setup a fake worker version ID
     os.environ["WORKER_VERSION_ID"] = "1234-{{ cookiecutter.slug }}"
+    # Setup a fake worker run ID
+    os.environ["ARKINDEX_WORKER_RUN_ID"] = "1234-{{ cookiecutter.slug }}"