Bugfix (#247)

* Fixed has no attribute 'json' * Fixed has no num_thread
tusen-ai · Oct 17, 2019 · 67c79f5 · 67c79f5
1 parent 6cb3a07
commit 67c79f5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/unittest/test_loader.py b/unittest/test_loader.py
@@ -23,7 +23,7 @@ def test_empty_v_loader(self):
             label_name=label_name,
             batch_size=1,
             shuffle=True,
-            num_thread=1,
+            num_worker=1,
             kv=mx.kvstore.create(pKv.kvstore)
         )
         with self.assertRaises(StopIteration):
@@ -43,7 +43,7 @@ def test_empty_h_loader(self):
             label_name=label_name,
             batch_size=1,
             shuffle=True,
-            num_thread=1,
+            num_worker=1,
             kv=mx.kvstore.create(pKv.kvstore)
         )
         with self.assertRaises(StopIteration):
@@ -63,7 +63,7 @@ def test_record_num(self):
             label_name=label_name,
             batch_size=batch_size,
             shuffle=True,
-            num_thread=1,
+            num_worker=1,
             kv=mx.kvstore.create(pKv.kvstore)
         )
 

diff --git a/utils/json_to_roidb.py b/utils/json_to_roidb.py
@@ -9,8 +9,8 @@
 def parse_argument():
     parser = argparse.ArgumentParser("Convert json gt to roidb")
     parser.add_argument("--json", type=str, required=True)
-    parser.parse_args()
-    return parser.json
+    args = parser.parse_args()
+    return args.json
 
 
 def json_to_roidb(json_path):