Run black to fix linting issues

training/test_train.py

@@ -12,7 +12,6 @@
 
 @pytest.fixture
 def model():
-
     # Load dataset
     data = pd.read_csv("dataset/sqli_dataset.csv")
 
@@ -26,15 +25,22 @@ def model():
     return {"tokenizer": tokenizer, "sqli_model": sqli_model}
 
 
-@pytest.mark.parametrize("sample",
+@pytest.mark.parametrize(
+    "sample",
     [
         ("select * from users where id='1' or 1=1--", [92.02]),
         ("select * from users", [0.077]),
         ("select * from users where id=10000", [14.83]),
         ("select '1' union select 'a'; -- -'", [99.99]),
-        ("select '' union select 'malicious php code' \g /var/www/test.php; -- -';", [99.99]),
-        ("select '' || pg_sleep((ascii((select 'a' limit 1)) - 32) / 2); -- -';", [99.99])
-    ]
+        (
+            "select '' union select 'malicious php code' \g /var/www/test.php; -- -';",
+            [99.99],
+        ),
+        (
+            "select '' || pg_sleep((ascii((select 'a' limit 1)) - 32) / 2); -- -';",
+            [99.99],
+        ),
+    ],
 )
 def test_sqli_model(model, sample):
     # Vectorize the sample
@@ -45,6 +51,4 @@ def test_sqli_model(model, sample):
     predictions = model["sqli_model"].predict(sample_vec)
 
     # Scale up to 100
-    assert predictions * 100 == pytest.approx(
-        np.array([sample[1]]), 0.1
-    )
+    assert predictions * 100 == pytest.approx(np.array([sample[1]]), 0.1)