Add tag and description to training methods

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,