From 609e1676c72dff498843da08de3e9ec6e7881afc Mon Sep 17 00:00:00 2001
From: NolanB <nboukachab@teklia.com>
Date: Mon, 5 Sep 2022 13:00:45 +0200
Subject: [PATCH] Modif with the review
---
arkindex_worker/worker/classification.py | 1 -
tests/test_elements_worker/test_entities.py | 1 +
2 files changed, 1 insertion(+), 1 deletion(-)
diff --git a/arkindex_worker/worker/classification.py b/arkindex_worker/worker/classification.py
index 54e09862..ee1e1b2d 100644
--- a/arkindex_worker/worker/classification.py
+++ b/arkindex_worker/worker/classification.py
@@ -36,7 +36,6 @@ class ClassificationMixin(object):
:param ml_class str: Name of the MLClass.
:returns str: ID of the retrieved or created MLClass.
"""
-
if not self.classes.get(self.corpus_id):
self.load_corpus_classes()
diff --git a/tests/test_elements_worker/test_entities.py b/tests/test_elements_worker/test_entities.py
index 41e53644..84963d79 100644
--- a/tests/test_elements_worker/test_entities.py
+++ b/tests/test_elements_worker/test_entities.py
@@ -94,6 +94,7 @@ def test_create_entity_wrong_corpus(monkeypatch, mock_elements_worker):
elt = Element({"id": "12341234-1234-1234-1234-123412341234"})
# Triggering an error on metas param, not giving corpus should work since
+ # ARKINDEX_CORPUS_ID environment variable is set on mock_elements_worker
with pytest.raises(AssertionError) as e:
mock_elements_worker.create_entity(
element=elt,
--
GitLab