From 56b64719ebec9c24d4c421f0b059640308e49ed8 Mon Sep 17 00:00:00 2001
From: David Luevano <55825613+luevano@users.noreply.github.com>
Date: Thu, 12 Dec 2019 23:38:35 -0700
Subject: Fix bug

---
 do_ml.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

(limited to 'do_ml.py')

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:
-- 
cgit v1.2.3-54-g00ecf