Load dataset from parent dir

gatewayd-io · Oct 17, 2024 · 1311692 · 1311692
1 parent 81e09aa
commit 1311692
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/training/test_train.py b/training/test_train.py
@@ -34,7 +34,12 @@
 )
 def model(request):
     # Load dataset
-    data = pd.read_csv(request.param["dataset"])
+    data = None
+    try:
+        data = pd.read_csv(request.param["dataset"])
+    except FileNotFoundError:
+        # Check if the dataset is in the parent directory
+        data = pd.read_csv("../" + request.param["dataset"])
 
     # Load TF model from SavedModel
     sqli_model = TFSMLayer(request.param["model_path"], call_endpoint="serving_default")