From 3b949429f66b19fefcfd10851e96c0be1514481a Mon Sep 17 00:00:00 2001 From: Andrey Velichkevich Date: Fri, 8 Sep 2023 19:10:55 +0100 Subject: [PATCH] Fix client creation in test --- sdk/python/test/e2e/test_e2e_mxjob.py | 2 +- sdk/python/test/e2e/test_e2e_tfjob.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/sdk/python/test/e2e/test_e2e_mxjob.py b/sdk/python/test/e2e/test_e2e_mxjob.py index 8b71569def..4de608ed16 100644 --- a/sdk/python/test/e2e/test_e2e_mxjob.py +++ b/sdk/python/test/e2e/test_e2e_mxjob.py @@ -43,7 +43,7 @@ logging.basicConfig(format="%(message)s") logging.getLogger().setLevel(logging.INFO) -TRAINING_CLIENT = TrainingClient(constants.MXJOB_KIND) +TRAINING_CLIENT = TrainingClient(job_kind=constants.MXJOB_KIND) JOB_NAME = "mxjob-mnist-ci-test" CONTAINER_NAME = "mxnet" GANG_SCHEDULER_NAME = os.getenv(TEST_GANG_SCHEDULER_NAME_ENV_KEY, "") diff --git a/sdk/python/test/e2e/test_e2e_tfjob.py b/sdk/python/test/e2e/test_e2e_tfjob.py index 889757b4f5..14bc032a4d 100644 --- a/sdk/python/test/e2e/test_e2e_tfjob.py +++ b/sdk/python/test/e2e/test_e2e_tfjob.py @@ -42,7 +42,7 @@ logging.basicConfig(format="%(message)s") logging.getLogger().setLevel(logging.INFO) -TRAINING_CLIENT = TrainingClient(constants.TFJOB_KIND) +TRAINING_CLIENT = TrainingClient(job_kind=constants.TFJOB_KIND) JOB_NAME = "tfjob-mnist-ci-test" CONTAINER_NAME = "tensorflow" GANG_SCHEDULER_NAME = os.getenv(TEST_GANG_SCHEDULER_NAME_ENV_KEY, "")