Skip to content

Commit b3de39c

Browse files
author
aditya tewari
committed
removing argv for test
1 parent 1f2c305 commit b3de39c

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

examples/lienet/task.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
#!/usr/bin/env python3
22
import argparse
33
import os
4-
import sys
54
import tensorflow as tf
65
import numpy as np
76

@@ -14,8 +13,8 @@
1413
VAL_SPLIT = 0.2
1514

1615

17-
def get_args(args):
18-
parser = argparse.ArgumentParser(args)
16+
def get_args():
17+
parser = argparse.ArgumentParser()
1918
parser.add_argument(
2019
'--job-dir', type=str, required=True, help='checkpoint dir'
2120
)
@@ -91,5 +90,4 @@ def train_and_evaluate(args):
9190

9291
if __name__ == "__main__":
9392
tf.get_logger().setLevel("INFO")
94-
argv = sys.argv[1:]
95-
train_and_evaluate(get_args(argv))
93+
train_and_evaluate(get_args())

0 commit comments

Comments
 (0)