wip

Author: jopemachine

src/ai/backend/manager/models/endpoint/row.py

@@ -864,9 +864,7 @@ def _to_deployment_info_with_revisions(
             model_revisions=list(model_revisions),
             current_revision_id=self.current_revision,
             deploying_revision_id=self.deploying_revision,
-            sub_step=DeploymentLifecycleSubStep(self.sub_step)
-            if self.sub_step is not None
-            else None,
+            sub_step=self.sub_step,
         )
 
     def build_revision_spec_from_endpoint(self) -> ModelRevisionSpec:
@@ -934,9 +932,7 @@ def _to_deployment_info_legacy(self) -> DeploymentInfo:
             model_revisions=[self.build_revision_spec_from_endpoint()],
             current_revision_id=self.current_revision,
             deploying_revision_id=self.deploying_revision,
-            sub_step=DeploymentLifecycleSubStep(self.sub_step)
-            if self.sub_step is not None
-            else None,
+            sub_step=self.sub_step,
         )