From 799e7f629fde8c7da403173ce2c1afb97e77e338 Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Mon, 3 May 2021 18:32:07 +0300 Subject: [PATCH] Fix PY2 compatibility --- clearml/backend_interface/task/models.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/clearml/backend_interface/task/models.py b/clearml/backend_interface/task/models.py index 2a051c7a..77a85141 100644 --- a/clearml/backend_interface/task/models.py +++ b/clearml/backend_interface/task/models.py @@ -1,6 +1,16 @@ import re import typing -from collections import UserDict, OrderedDict, UserList +from collections import OrderedDict + +try: + from collections import UserList +except ImportError: + UserList = list + +try: + from collections import UserDict +except ImportError: + UserDict = dict from clearml.backend_api import Session from clearml.backend_api.services import models