fix a bug in export_coco_as_yolov5 (#262)

* fix a bug in `export_coco_as_yolov5` * update to v0.8.8
obss · Nov 2, 2021 · 9c634e9 · 9c634e9
1 parent 91e8b80
commit 9c634e9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/sahi/__init__.py b/sahi/__init__.py
@@ -1 +1 @@
-__version__ = "0.8.7"
+__version__ = "0.8.8"
diff --git a/sahi/utils/coco.py b/sahi/utils/coco.py
@@ -2230,9 +2230,7 @@ def export_coco_as_yolov5(
         ValueError("'train_coco' have to be provided")
 
     # check train_split_rate
-    if 0 < train_split_rate and train_split_rate < 1:
-        split_mode = "SPLIT"
-    else:
+    if split_mode and not (0 < train_split_rate < 1):
         ValueError("train_split_rate cannot be <0 or >1")
 
     # split dataset