Skip to content

Commit a47799b

Browse files
carusytecarusyte
committed
fix _parseArgs issue
1 parent 758a60b commit a47799b

File tree

4 files changed

+49
-49
lines changed

4 files changed

+49
-49
lines changed

‎corl/wc_test/test11.py

+17-17
Original file line numberDiff line numberDiff line change
@@ -39,39 +39,39 @@
3939

4040
def parseArgs():
4141
parser = argparse.ArgumentParser()
42-
parser._parseArgs('--ds', type=str, help='datasource. such as file, db, or BigQuery.',
42+
parser.add_argument('--ds', type=str, help='datasource. such as file, db, or BigQuery.',
4343
default='db')
44-
parser._parseArgs('--dir', type=str,
44+
parser.add_argument('--dir', type=str,
4545
help='directory path for training and test set.')
46-
parser._parseArgs('--parallel', type=int, help='database operation parallel level',
46+
parser.add_argument('--parallel', type=int, help='database operation parallel level',
4747
default=multiprocessing.cpu_count())
48-
parser._parseArgs('--prefetch', type=int, help='dataset prefetch batches',
48+
parser.add_argument('--prefetch', type=int, help='dataset prefetch batches',
4949
default=2)
50-
parser._parseArgs('--db_host', type=str, help='database host address',
50+
parser.add_argument('--db_host', type=str, help='database host address',
5151
default=None)
52-
parser._parseArgs('--db_port', type=int, help='database listening port',
52+
parser.add_argument('--db_port', type=int, help='database listening port',
5353
default=None)
54-
parser._parseArgs('--db_pwd', type=str, help='database password',
54+
parser.add_argument('--db_pwd', type=str, help='database password',
5555
default=None)
56-
parser._parseArgs('--vset', type=int, help='validation set number',
56+
parser.add_argument('--vset', type=int, help='validation set number',
5757
default=None)
58-
parser._parseArgs('--db_pool', type=int, help='database connection pool size',
58+
parser.add_argument('--db_pool', type=int, help='database connection pool size',
5959
default=multiprocessing.cpu_count())
60-
parser._parseArgs('--start', type=int, help='start training at specified batch no',
60+
parser.add_argument('--start', type=int, help='start training at specified batch no',
6161
default=None)
62-
parser._parseArgs('--vol_size', type=int, help='volume size for the dataset storage sub-folder',
62+
parser.add_argument('--vol_size', type=int, help='volume size for the dataset storage sub-folder',
6363
default=None)
64-
parser._parseArgs('--gpu_grow_mem', dest='gpu_grow_mem', action='store_true',
64+
parser.add_argument('--gpu_grow_mem', dest='gpu_grow_mem', action='store_true',
6565
help='allow gpu to allocate mem dynamically at runtime.')
66-
parser._parseArgs('--trace', dest='trace', action='store_true',
66+
parser.add_argument('--trace', dest='trace', action='store_true',
6767
help='record full trace in validation step.')
68-
parser._parseArgs('--profile', dest='profile', action='store_true',
68+
parser.add_argument('--profile', dest='profile', action='store_true',
6969
help='profile CG execution.')
70-
parser._parseArgs('--skip_init_test', dest='skip_init_test', action='store_true',
70+
parser.add_argument('--skip_init_test', dest='skip_init_test', action='store_true',
7171
help='whether to skip the initial test.')
72-
parser._parseArgs('--log_device', dest='log_device', action='store_true',
72+
parser.add_argument('--log_device', dest='log_device', action='store_true',
7373
help='record device info such as CPU and GPU in tensorboard.')
74-
parser._parseArgs(
74+
parser.add_argument(
7575
'--restart', help='restart training', action='store_true')
7676
return parser.parse_args()
7777

‎corl/wc_test/test6.py

+10-10
Original file line numberDiff line numberDiff line change
@@ -33,25 +33,25 @@
3333

3434
def parseArgs():
3535
parser = argparse.ArgumentParser()
36-
parser._parseArgs('--ds', type=str, help='datasource. such as db, or BigQuery.',
36+
parser.add_argument('--ds', type=str, help='datasource. such as db, or BigQuery.',
3737
default='db')
38-
parser._parseArgs('--parallel', type=int, help='database operation parallel level',
38+
parser.add_argument('--parallel', type=int, help='database operation parallel level',
3939
default=multiprocessing.cpu_count())
40-
parser._parseArgs('--prefetch', type=int, help='dataset prefetch batches',
40+
parser.add_argument('--prefetch', type=int, help='dataset prefetch batches',
4141
default=2)
42-
parser._parseArgs('--db_host', type=str, help='database host address',
42+
parser.add_argument('--db_host', type=str, help='database host address',
4343
default=None)
44-
parser._parseArgs('--db_port', type=int, help='database listening port',
44+
parser.add_argument('--db_port', type=int, help='database listening port',
4545
default=None)
46-
parser._parseArgs('--db_pwd', type=str, help='database password',
46+
parser.add_argument('--db_pwd', type=str, help='database password',
4747
default=None)
48-
parser._parseArgs('--vset', type=int, help='validation set number',
48+
parser.add_argument('--vset', type=int, help='validation set number',
4949
default=None)
50-
parser._parseArgs('--db_pool', type=int, help='database connection pool size',
50+
parser.add_argument('--db_pool', type=int, help='database connection pool size',
5151
default=multiprocessing.cpu_count())
52-
parser._parseArgs('--start', type=int, help='start training at specified batch no',
52+
parser.add_argument('--start', type=int, help='start training at specified batch no',
5353
default=None)
54-
parser._parseArgs(
54+
parser.add_argument(
5555
'--restart', help='restart training', action='store_true')
5656
return parser.parse_args()
5757

‎corl/wcc/wcc_infer.py

+11-11
Original file line numberDiff line numberDiff line change
@@ -33,28 +33,28 @@
3333
def parseArgs():
3434
parser = argparse.ArgumentParser()
3535
req = parser.add_argument_group('required named arguments')
36-
req._parseArgs('-m', '--model', type=str,
36+
req.add_argument('-m', '--model', type=str,
3737
help='path to the saved model', required=True)
38-
req._parseArgs('-r', '--rbase', type=str,
38+
req.add_argument('-r', '--rbase', type=str,
3939
help='gcs remote directory path for inference file', required=True)
40-
req._parseArgs('-p', '--path', type=str,
40+
req.add_argument('-p', '--path', type=str,
4141
help='gcs remote directory path for inference result files', required=True)
4242

43-
parser._parseArgs('-d', '--del_used', dest='del_used', default=False,
43+
parser.add_argument('-d', '--del_used', dest='del_used', default=False,
4444
action='store_true', help='delete used inference files.')
45-
parser._parseArgs('--project', dest='project', type=str,
45+
parser.add_argument('--project', dest='project', type=str,
4646
help='gcs remote directory path for inference result files')
47-
parser._parseArgs('-f', '--prefetch', type=int,
47+
parser.add_argument('-f', '--prefetch', type=int,
4848
help='dataset prefetch batches', default=2)
49-
parser._parseArgs('-b', '--batch', type=int, dest="batch",
49+
parser.add_argument('-b', '--batch', type=int, dest="batch",
5050
help='inference result file batch size', default=128)
51-
parser._parseArgs('-g', '--gpu_grow_mem', dest='gpu_grow_mem', default=False,
51+
parser.add_argument('-g', '--gpu_grow_mem', dest='gpu_grow_mem', default=False,
5252
action='store_true', help='allow gpu to allocate mem dynamically at runtime.')
53-
parser._parseArgs('--trace', dest='trace', action='store_true', default=False,
53+
parser.add_argument('--trace', dest='trace', action='store_true', default=False,
5454
help='record full trace in validation step.')
55-
parser._parseArgs('--profile', dest='profile', action='store_true', default=False,
55+
parser.add_argument('--profile', dest='profile', action='store_true', default=False,
5656
help='profile CG execution.')
57-
parser._parseArgs('--log_device', dest='log_device', action='store_true', default=False,
57+
parser.add_argument('--log_device', dest='log_device', action='store_true', default=False,
5858
help='record device info such as CPU and GPU in tensorboard.')
5959
return parser.parse_args()
6060

‎dataset/main.py

+11-11
Original file line numberDiff line numberDiff line change
@@ -7,27 +7,27 @@
77

88
def parseArgs():
99
parser = argparse.ArgumentParser()
10-
parser._parseArgs('--table', nargs='+', type=str, help='database tables to be exported.',
10+
parser.add_argument('--table', nargs='+', type=str, help='database tables to be exported.',
1111
default=None)
12-
parser._parseArgs('--fields', nargs='+', type=str, help='fields to be exported.',
12+
parser.add_argument('--fields', nargs='+', type=str, help='fields to be exported.',
1313
default=None)
14-
parser._parseArgs('--dest', type=str, help='destination folder.',
14+
parser.add_argument('--dest', type=str, help='destination folder.',
1515
default=None)
16-
parser._parseArgs('--format', type=str, help='exported file format (avro, json).',
16+
parser.add_argument('--format', type=str, help='exported file format (avro, json).',
1717
default='avro')
18-
parser._parseArgs('--zip', type=bool, help='compress exported file.',
18+
parser.add_argument('--zip', type=bool, help='compress exported file.',
1919
default=False)
20-
parser._parseArgs('--flags', nargs='+', type=str, help='export sets with the specified flag (TR/TS).',
20+
parser.add_argument('--flags', nargs='+', type=str, help='export sets with the specified flag (TR/TS).',
2121
default=None)
22-
parser._parseArgs('--start', type=int, help='export sets starting with the specified batch number.',
22+
parser.add_argument('--start', type=int, help='export sets starting with the specified batch number.',
2323
default=None)
24-
parser._parseArgs('--end', type=int, help='export sets ending with the specified batch number.',
24+
parser.add_argument('--end', type=int, help='export sets ending with the specified batch number.',
2525
default=None)
26-
parser._parseArgs('--vol_size', type=int, help='volume size of each sub-folder in the destination folder.',
26+
parser.add_argument('--vol_size', type=int, help='volume size of each sub-folder in the destination folder.',
2727
default=None)
28-
parser._parseArgs('--parallel', type=int, help='parallelization level.',
28+
parser.add_argument('--parallel', type=int, help='parallelization level.',
2929
default=None)
30-
parser._parseArgs('options', nargs=argparse.REMAINDER)
30+
parser.add_argument('options', nargs=argparse.REMAINDER)
3131
return parser.parse_args()
3232

3333

0 commit comments

Comments
 (0)