Fix lintin

arkindex_worker/worker/entity.py

@@ -40,9 +40,6 @@ class EntityMixin(object):
         :type element: Element or CachedElement
         :param name str: Name of the entity.
         :param type EntityType: Type of the entity.
-        :param corpus: UUID of the corpus to create an entity on, or None to use the
-           value of the ``ARKINDEX_CORPUS_ID`` environment variable.
-        :type corpus: str or None
         """
         assert element and isinstance(
             element, (Element, CachedElement)

tests/test_elements_worker/test_classifications.py

@@ -36,7 +36,10 @@ def test_get_ml_class_id_load_classes(responses, mock_elements_worker):
     assert [
         (call.request.method, call.request.url) for call in responses.calls
     ] == BASE_API_CALLS + [
-        ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
+        (
+            "GET",
+            f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        ),
     ]
     assert mock_elements_worker.classes == {
         "12341234-1234-1234-1234-123412341234": {"good": "0000"}
@@ -46,9 +49,9 @@ def test_get_ml_class_id_load_classes(responses, mock_elements_worker):
 
 def test_get_ml_class_id_inexistant_class(mock_elements_worker, responses):
     # A missing class is now created automatically
-    mock_elements_worker.corpus_id  = "12341234-1234-1234-1234-123412341234"
+    mock_elements_worker.corpus_id = "12341234-1234-1234-1234-123412341234"
     mock_elements_worker.classes = {
-         "12341234-1234-1234-1234-123412341234": {"good": "0000"}
+        "12341234-1234-1234-1234-123412341234": {"good": "0000"}
     }
 
     responses.add(
@@ -145,9 +148,18 @@ def test_get_ml_class_reload(responses, mock_elements_worker):
     assert [
         (call.request.method, call.request.url) for call in responses.calls
     ] == BASE_API_CALLS + [
-        ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
-        ("POST", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
-        ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
+        (
+            "GET",
+            f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        ),
+        (
+            "POST",
+            f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        ),
+        (
+            "GET",
+            f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        ),
     ]
 
 

tests/test_elements_worker/test_elements.py

@@ -289,11 +289,26 @@ def test_load_corpus_classes_api_error(responses, mock_elements_worker):
         (call.request.method, call.request.url) for call in responses.calls
     ] == BASE_API_CALLS + [
         # We do 5 retries
-        ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
-        ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
-        ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
-        ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
-        ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
+        (
+            "GET",
+            f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        ),
+        (
+            "GET",
+            f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        ),
+        (
+            "GET",
+            f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        ),
+        (
+            "GET",
+            f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        ),
+        (
+            "GET",
+            f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        ),
     ]
     assert not mock_elements_worker.classes
 
@@ -331,7 +346,10 @@ def test_load_corpus_classes(responses, mock_elements_worker):
     assert [
         (call.request.method, call.request.url) for call in responses.calls
     ] == BASE_API_CALLS + [
-        ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
+        (
+            "GET",
+            f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        ),
     ]
     assert mock_elements_worker.classes == {
         "12341234-1234-1234-1234-123412341234": {