From d63fedd2971bbe725557923a901c5bc8bf072192 Mon Sep 17 00:00:00 2001 From: CptCaptain Date: Mon, 29 Nov 2021 18:20:39 +0100 Subject: [PATCH] fix off-by-one error --- create_syn_data.sh | 2 +- data/create_syn_data.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/create_syn_data.sh b/create_syn_data.sh index 35573b2..5904d47 100755 --- a/create_syn_data.sh +++ b/create_syn_data.sh @@ -4,4 +4,4 @@ cd data/lcn python setup.py build_ext --inplace cd ../ -python create_syn_data.py $2 +python create_syn_data.py $1 diff --git a/data/create_syn_data.py b/data/create_syn_data.py index 809206e..2615235 100644 --- a/data/create_syn_data.py +++ b/data/create_syn_data.py @@ -228,10 +228,10 @@ if __name__ == '__main__': out_root.mkdir(parents=True, exist_ok=True) start = 0 - if len(sys.argv) >= 2 and isinstance(sys.argv[2], int): - start = sys.argv[2] + if len(sys.argv) >= 2 and isinstance(sys.argv[1], int): + start = sys.argv[1] else: - if sys.argv[2] == '--resume': + if sys.argv[1] == '--resume': try: start = max([int(dir) for dir in os.listdir(out_root) if str.isdigit(dir)]) or 0 except: