From 206d86ec5b6f34b0a4856249164af755f2f54cf3 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Mon, 31 May 2021 15:43:02 +0300
Subject: [PATCH] Support DummyModel configuration (Slack
 https://clearml.slack.com/archives/CTK20V944/p1621469235085400)

---
 clearml/model.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/clearml/model.py b/clearml/model.py
index 5aaa7d29..33e7f5a6 100644
--- a/clearml/model.py
+++ b/clearml/model.py
@@ -1104,9 +1104,11 @@ class OutputModel(BaseModel):
         # noinspection PyProtectedMember
         if not self._task._get_model_config_text():
             # noinspection PyProtectedMember
-            task._set_model_config(config_text=model.model_design)
+            task._set_model_config(
+                config_text=model.model_design if hasattr(model, "model_design") else model.design.get("design", "")
+            )
         if not self._task.get_labels_enumeration():
-            task.set_model_label_enumeration(model.data.labels)
+            task.set_model_label_enumeration(model.data.labels if hasattr(model, "data") else model.labels)
 
         if self._base_model:
             self._base_model.update_for_task(