chore: Update get_trial_by_model_and_input to handle JSON input URL

c3sr · Jul 24, 2024 · 2c97869 · 2c97869
1 parent 95fc212
commit 2c97869
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python_api/db.py b/python_api/db.py
@@ -48,11 +48,12 @@ def create_expriement( cur, conn):
 
 
 def get_trial_by_model_and_input(model_id, input_url):
-
+    input_url=json.loads(input_url)
+    input_url=input_url["url"]
     input_query = """
             SELECT trial_id
             FROM trial_inputs
-            WHERE url = %s
+            WHERE %s in url
         """
 
     # Main query to get trial details