From 87e4247506229036468a21e523d6ac97370cf06f Mon Sep 17 00:00:00 2001 From: YuanTingHsieh Date: Sun, 14 Apr 2024 18:26:48 -0700 Subject: [PATCH] Fix tf integration test --- .../hello-tf2/jobs/hello-tf2/app/custom/trainer.py | 2 +- tests/integration_test/src/utils.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/examples/hello-world/hello-tf2/jobs/hello-tf2/app/custom/trainer.py b/examples/hello-world/hello-tf2/jobs/hello-tf2/app/custom/trainer.py index bbb8c17b9b..25725687d4 100644 --- a/examples/hello-world/hello-tf2/jobs/hello-tf2/app/custom/trainer.py +++ b/examples/hello-world/hello-tf2/jobs/hello-tf2/app/custom/trainer.py @@ -14,7 +14,7 @@ import os -os.enviro["TF_FORCE_GPU_ALLOW_GROWTH"] = "true" +os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true" import numpy as np import tensorflow as tf diff --git a/tests/integration_test/src/utils.py b/tests/integration_test/src/utils.py index 0a945607c9..d80f34b13b 100644 --- a/tests/integration_test/src/utils.py +++ b/tests/integration_test/src/utils.py @@ -195,6 +195,10 @@ def check_client_status_ready(response: dict) -> bool: if "client_statuses" not in response["details"]: return False + for row in response["details"]["client_statuses"][1:]: + if row[3] == "No Reply": + return False + return True