Skip to content

Commit 5dff9ed

Browse files
Fix model serving integration test (#2460)
1 parent 6cf4af2 commit 5dff9ed

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

internal/acceptance/model_serving_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,9 +83,9 @@ func TestAccModelServing(t *testing.T) {
8383
client = MlflowClient()
8484
client.create_model_version(name="%[1]s-model", source=source, run_id=run1_id)
8585
client.create_model_version(name="%[1]s-model", source=source, run_id=run1_id)
86-
while client.get_model_version(name="%[1]s-model", version="1").getStatus() != ModelRegistry.ModelVersionStatus.READY:
86+
while client.get_model_version(name="%[1]s-model", version="1").status != "READY":
8787
time.sleep(10)
88-
while client.get_model_version(name="%[1]s-model", version="2").getStatus() != ModelRegistry.ModelVersionStatus.READY:
88+
while client.get_model_version(name="%[1]s-model", version="2").status != "READY":
8989
time.sleep(10)
9090
`, name))
9191
if err != nil {

0 commit comments

Comments
 (0)