From bcf90852112e2874d8e34223e4633ca4e9fc3394 Mon Sep 17 00:00:00 2001
From: NolanB <nboukachab@teklia.com>
Date: Tue, 30 Aug 2022 11:50:34 +0200
Subject: [PATCH] Add tag and description to training methods

---
 tests/test_elements_worker/test_training.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/test_elements_worker/test_training.py b/tests/test_elements_worker/test_training.py
index 9b6f014a..73970081 100644
--- a/tests/test_elements_worker/test_training.py
+++ b/tests/test_elements_worker/test_training.py
@@ -56,6 +56,8 @@ def test_create_model_version(tag, description):
     model_hash = "hash"
     archive_hash = "archive_hash"
     size = "30"
+    tag = "tag"
+    description = "description"
     model_version_details = {
         "id": model_version_id,
         "model_id": model_id,
-- 
GitLab