From d2821a657452a030eae7085f323b64b2f6e44676 Mon Sep 17 00:00:00 2001
From: Erwan Rouchet <rouchet@teklia.com>
Date: Wed, 31 Mar 2021 15:31:15 +0200
Subject: [PATCH] Replace PONOS_DATA_DIR with the existing PONOS_DATA

---
 arkindex_worker/worker/base.py | 2 +-
 tests/test_merge.py            | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/arkindex_worker/worker/base.py b/arkindex_worker/worker/base.py
index c3a08f1d..78cbb8d6 100644
--- a/arkindex_worker/worker/base.py
+++ b/arkindex_worker/worker/base.py
@@ -145,7 +145,7 @@ class BaseWorker(object):
             merge_parents_cache(
                 task["parents"],
                 self.cache_path,
-                data_dir=os.environ.get("PONOS_DATA_DIR", "/data"),
+                data_dir=os.environ.get("PONOS_DATA", "/data"),
                 chunk=os.environ.get("ARKINDEX_TASK_CHUNK"),
             )
 
diff --git a/tests/test_merge.py b/tests/test_merge.py
index 344f067c..a2061622 100644
--- a/tests/test_merge.py
+++ b/tests/test_merge.py
@@ -151,7 +151,7 @@ def test_merge_from_worker(
     assert CachedTranscription.select().count() == 0
 
     # Configure worker with a specific data directory
-    monkeypatch.setenv("PONOS_DATA_DIR", str(tmpdir))
+    monkeypatch.setenv("PONOS_DATA", str(tmpdir))
     mock_base_worker_with_cache.configure()
 
     # Then we have 2 elements and a transcription
-- 
GitLab