Skip to content

Commit af4f29f

Browse files
map copy status to model status in CustomFormModelInfo (#11666)
1 parent 36072c1 commit af4f29f

File tree

3 files changed

+4
-2
lines changed

3 files changed

+4
-2
lines changed

sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_models.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -867,6 +867,8 @@ def __init__(self, **kwargs):
867867

868868
@classmethod
869869
def _from_generated(cls, model, model_id=None):
870+
if model.status == "succeeded": # map copy status to model status
871+
model.status = "ready"
870872
return cls(
871873
model_id=model_id if model_id else model.model_id,
872874
status=model.status,

sdk/formrecognizer/azure-ai-formrecognizer/tests/test_copy_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def test_copy_model_successful(self, client, container_sas_url, location, resour
4343

4444
copied_model = client.get_custom_model(copy.model_id)
4545

46-
self.assertEqual(copy.status, "succeeded")
46+
self.assertEqual(copy.status, "ready")
4747
self.assertIsNotNone(copy.requested_on)
4848
self.assertIsNotNone(copy.completed_on)
4949
self.assertEqual(target["modelId"], copy.model_id)

sdk/formrecognizer/azure-ai-formrecognizer/tests/test_copy_model_async.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ async def test_copy_model_successful(self, client, container_sas_url, location,
4242

4343
copied_model = await client.get_custom_model(copy.model_id)
4444

45-
self.assertEqual(copy.status, "succeeded")
45+
self.assertEqual(copy.status, "ready")
4646
self.assertIsNotNone(copy.requested_on)
4747
self.assertIsNotNone(copy.completed_on)
4848
self.assertEqual(target["modelId"], copy.model_id)

0 commit comments

Comments
 (0)