From b5c6c3ae4c313420053ffe5ff4353aef0d9b7ef8 Mon Sep 17 00:00:00 2001 From: Sourcery AI <> Date: Wed, 7 Apr 2021 14:30:02 +0000 Subject: [PATCH] 'Refactored by Sourcery' --- dataikuapi/dss/modelevaluationstore.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dataikuapi/dss/modelevaluationstore.py b/dataikuapi/dss/modelevaluationstore.py index 9c247d46..d571a0a2 100644 --- a/dataikuapi/dss/modelevaluationstore.py +++ b/dataikuapi/dss/modelevaluationstore.py @@ -118,7 +118,7 @@ def list_model_evaluations(self, as_type=None): :rtype: list """ items = self.client._perform_json("GET", "/projects/%s/modelevaluationstores/%s/runs/" % (self.project_key, self.mes_id)) - if as_type == "objects" or as_type == "object": + if as_type in ["objects", "object"]: return [DSSModelEvaluation(self, item["ref"]["runId"]) for item in items] else: return items