From 7b5009da862bba5baae1d23cd916e75b38a84eb1 Mon Sep 17 00:00:00 2001 From: Yukuo Cen Date: Thu, 6 Jan 2022 21:55:50 +0800 Subject: [PATCH] [Bugfix] Fix copying args (#326) --- cogdl/experiments.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/cogdl/experiments.py b/cogdl/experiments.py index 224fd7b1..3b879018 100644 --- a/cogdl/experiments.py +++ b/cogdl/experiments.py @@ -223,9 +223,11 @@ def gen_variants(**items): def variant_args_generator(args, variants): """Form variants as group with size of num_workers""" - for variant in variants: + for idx, variant in enumerate(variants): args.dataset, args.model, args.seed, args.split = variant - yield copy.deepcopy(args) + if idx < len(variants) - 1: + yield copy.deepcopy(args) + yield args def output_results(results_dict, tablefmt="github"):