Skip to content

Commit b296657

Browse files
authored
Remove unnecessary Worker param (#2499)
* Remove unnecessary Worker param when in test, it will throw warning log. like following [2021-02-02 18:02:49,547] [WARNING] [args.py:239:parse_worker_args] Unknown arguments: ['--worker_id', '0'] * Update worker_ps_interaction_test.py
1 parent e60551b commit b296657

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

elasticdl/python/tests/worker_ps_interaction_test.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -71,8 +71,6 @@ def _create_worker(self, worker_num):
7171
tf.keras.backend.clear_session()
7272
tf.random.set_seed(22)
7373
arguments = [
74-
"--worker_id",
75-
i,
7674
"--job_type",
7775
elasticai_api_pb2.TRAINING,
7876
"--minibatch_size",
@@ -156,8 +154,6 @@ def test_worker_pull_embedding(self):
156154
model_def = "mnist.mnist_functional_api.custom_model"
157155
self._create_pserver(model_def, 2)
158156
arguments = [
159-
"--worker_id",
160-
0,
161157
"--job_type",
162158
elasticai_api_pb2.TRAINING,
163159
"--minibatch_size",
@@ -210,8 +206,6 @@ def test_compare_onebatch_train(self):
210206
images, labels = get_random_batch(self._batch_size)
211207
# TODO(yunjian.lmh): test optimizer wrapper
212208
arguments = [
213-
"--worker_id",
214-
0,
215209
"--job_type",
216210
elasticai_api_pb2.TRAINING,
217211
"--minibatch_size",
@@ -371,8 +365,6 @@ def test_restart_ps(self):
371365
for w in range(2):
372366
self._reset_pserver()
373367
arguments = [
374-
"--worker_id",
375-
0,
376368
"--job_type",
377369
elasticai_api_pb2.TRAINING,
378370
"--minibatch_size",

0 commit comments

Comments
 (0)