diff --git a/src/superannotate/lib/core/usecases/projects.py b/src/superannotate/lib/core/usecases/projects.py index 6963e49d..3877c5e5 100644 --- a/src/superannotate/lib/core/usecases/projects.py +++ b/src/superannotate/lib/core/usecases/projects.py @@ -497,7 +497,13 @@ def execute(self): step_setting = next( (i for i in project_settings if i.attribute == "WorkflowType"), None ) - if step_setting.value == constants.StepsType.BASIC.value: + if step_setting.value == constants.StepsType.KEYPOINT.value: + self._response.data = ( + self._service_provider.projects.list_keypoint_steps( + self._project + ).data["steps"] + ) + else: data = [] steps = self._service_provider.projects.list_steps(self._project).data for step in steps: @@ -511,12 +517,6 @@ def execute(self): break data.append(step_data) self._response.data = data - else: - self._response.data = ( - self._service_provider.projects.list_keypoint_steps( - self._project - ).data["steps"] - ) return self._response