This commit is contained in:
allegroai 2021-04-28 13:11:16 +03:00
parent 84025a75b2
commit 19c88f93d0
2 changed files with 5 additions and 5 deletions

View File

@ -8,7 +8,7 @@ from clearml.backend_api.services import models
class ModelsList(UserList):
def __init__(self, models_dict):
# type: (typing.OrderedDict["clearml.Model"]) -> None
# type: (typing.OrderedDict["clearml.Model"]) -> None # noqa: F821
self._models = models_dict
super(ModelsList, self).__init__(models_dict.values())
@ -38,14 +38,14 @@ class TaskModels(UserDict):
return self._output
def __init__(self, task):
# type: ("clearml.Task") -> None
# type: ("clearml.Task") -> None # noqa: F821
self._input = self._get_input_models(task)
self._output = self._get_output_models(task)
super(TaskModels, self).__init__({"input": self._input, "output": self._output})
def _get_input_models(self, task):
# type: ("clearml.Task") -> ModelsList
# type: ("clearml.Task") -> ModelsList # noqa: F821
if Session.check_min_api_version("2.13"):
parsed_ids = list(task.input_models_id.values())
@ -88,7 +88,7 @@ class TaskModels(UserDict):
@staticmethod
def _get_output_models(task):
# type: ("clearml.Task") -> ModelsList
# type: ("clearml.Task") -> ModelsList # noqa: F821
res = task.send(
models.GetAllRequest(

View File

@ -418,7 +418,7 @@ class PatchedMatplotlib:
lines_ = plotly_renderer.plotly_fig['data']
half_mark = len(lines_)//2
if len(lines_) % 2 == 0 and \
all(l for l in lines_[half_mark:] if not l.get('x') and not l.get('y')):
all(ln for ln in lines_[half_mark:] if not ln.get('x') and not ln.get('y')):
for i, line in enumerate(lines_[:half_mark]):
line['name'] = lines_[i+half_mark].get('name')
except Exception: