Merge pull request #8 from GreatV/bug_fixed

fix bug when test_size not set

src/labelme2yolo/__about__.py

@@ -2,4 +2,4 @@
 #
 # SPDX-License-Identifier: MIT
 
-__version__ = '0.0.6'
+__version__ = '0.0.7'

src/labelme2yolo/l2y.py

@@ -179,6 +179,8 @@ class Labelme2YOLO(object):
                                                 test_size=val_size)
         tmp_train_len = len(train_idxs)
         test_idxs = []
+        if test_size is None:
+            test_size = 0.0
         if test_size > 1e-8:
             train_idxs, test_idxs = train_test_split(
                 range(tmp_train_len), test_size=test_size / (1 - val_size))