From 99d963316d4f4cde6fa405a78f791bcbea2b6de0 Mon Sep 17 00:00:00 2001
From: Wang Xin <xinwang614@gmail.com>
Date: Mon, 1 Aug 2022 15:51:49 +0800
Subject: [PATCH] reformat code

---
 src/labelme2yolo/__main__.py     |  2 +-
 src/labelme2yolo/cli/__init__.py | 16 ++++++++--------
 2 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/src/labelme2yolo/__main__.py b/src/labelme2yolo/__main__.py
index beda96e..856c948 100644
--- a/src/labelme2yolo/__main__.py
+++ b/src/labelme2yolo/__main__.py
@@ -6,4 +6,4 @@ import sys
 if __name__ == '__main__':
     from .cli import run
 
-    sys.exit(run())
\ No newline at end of file
+    sys.exit(run())
diff --git a/src/labelme2yolo/cli/__init__.py b/src/labelme2yolo/cli/__init__.py
index 18e2d70..3c821bb 100644
--- a/src/labelme2yolo/cli/__init__.py
+++ b/src/labelme2yolo/cli/__init__.py
@@ -9,23 +9,23 @@ from labelme2yolo.l2y import Labelme2YOLO
 
 def run():
     parser = argparse.ArgumentParser("labelme2yolo")
-    parser.add_argument('--json_dir',type=str,
+    parser.add_argument('--json_dir', type=str,
                         help='Please input the path of the labelme json files.')
-    parser.add_argument('--val_size',type=float, nargs='?', default=None,
+    parser.add_argument('--val_size', type=float, nargs='?', default=None,
                         help='Please input the validation dataset size, for example 0.1 ')
-    parser.add_argument('--test_size',type=float, nargs='?', default=0.0,
+    parser.add_argument('--test_size', type=float, nargs='?', default=0.0,
                         help='Please input the validation dataset size, for example 0.1 ')
-    parser.add_argument('--json_name',type=str, nargs='?', default=None,
+    parser.add_argument('--json_name', type=str, nargs='?', default=None,
                         help='If you put json name, it would convert only one json file to YOLO.')
     args = parser.parse_args()
-    
+
     if not args.json_dir:
         parser.print_help(sys.stderr)
         sys.exit(1)
-    
+
     convertor = Labelme2YOLO(args.json_dir)
-    
+
     if args.json_name is None:
         convertor.convert(val_size=args.val_size, test_size=args.test_size)
     else:
-        convertor.convert_one(args.json_name)
\ No newline at end of file
+        convertor.convert_one(args.json_name)