Skip to content

Commit 12f3fd7

Browse files
committed
fix ut. test=develop
1 parent 478c24e commit 12f3fd7

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

python/paddle/distributed/fleet/base/fleet_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -692,7 +692,7 @@ def load_model(self, path, mode):
692692

693693
@is_non_distributed_check
694694
@inited_runtime_handler
695-
def run_heter_worker(self, dataset=None):
695+
def run_heter_worker(self, dataset):
696696
"""
697697
run_heter_worker will run heter trainer main program with executor.
698698

python/paddle/fluid/tests/unittests/dist_fleet_heter_pipeline_ctr.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ def do_dataset_heter_training(self, fleet):
194194
batch_size = 128
195195

196196
pass_start = time.time()
197-
fleet.run_heter_worker()
197+
fleet.run_heter_worker(dataset=None)
198198
pass_time = time.time() - pass_start
199199
print("do_dataset_heter_training done. using time {}".format(pass_time))
200200

0 commit comments

Comments
 (0)