fix(clustering)

science-engineering-art · Dec 21, 2022 · ac94703 · ac94703
1 parent 23c2466
commit ac94703
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -64,7 +64,10 @@ async def root(model: str, dataset: str, query: str, pag: int = 1):
     result = []  
     pag -= 1
 
-    if model in feedback and query in feedback[model][dataset].queries:
+    print(models[model])
+
+    if model in feedback and dataset in feedback[model] \
+        and query in feedback[model][dataset].queries:
         ranking = feedback[model][dataset].search(query)
     else:
         if type(models[model][dataset]) == str: