Update prediction scores after retraining the model

training/test_train.py

@@ -47,10 +47,10 @@ def model(request):
 @pytest.mark.parametrize(
     "sample",
     [
-        ("select * from users where id=1 or 1=1;", [99.99, 62.90]),
-        ("select * from users where id='1' or 1=1--", [92.02, 62.90]),
-        ("select * from users", [0.077, 0.011]),
-        ("select * from users where id=10000", [14.83, 1.319]),
+        ("select * from users where id=1 or 1=1;", [99.99, 0.11]),
+        ("select * from users where id='1' or 1=1--", [92.02, 0.11]),
+        ("select * from users", [0.077, 0.0044]),
+        ("select * from users where id=10000", [14.83, 0.10]),
         ("select '1' union select 'a'; -- -'", [99.99, 99.99]),
         (
             "select '' union select 'malicious php code' \g /var/www/test.php; -- -';",
@@ -71,6 +71,7 @@ def test_sqli_model(model, sample):
     predictions = model["sqli_model"].predict(sample_vec)
 
     # Scale up to 100
+    print(predictions * 100)  # Debugging purposes (prints on error)
     assert predictions * 100 == pytest.approx(
         np.array([[sample[1][model["index"]]]]), 0.1
     )