Fix bug

1 files changed, 2 insertions, 3 deletions

diff --git a/do_ml.py b/do_ml.py
index a424a610a..c88533e68 100644
--- a/do_ml.py
+++ b/do_ml.py
@@ -68,10 +68,9 @@ def do_ml(desc_data,
         return None
 
     if not test_size:
-        if d_len - training_size > 1500:
+        test_size = d_len - training_size
+        if test_size > 1500:
             test_size = 1500
-        else:
-            test_size = d_len - training_size
 
     tic = time.perf_counter()
     if show_msgs: