fixed theutil format issue

Jaseci-Labs · Mar 25, 2024 · 3b8c4d4 · 3b8c4d4
1 parent 10b874f
commit 3b8c4d4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/components/utils.jac b/src/components/utils.jac
@@ -62,7 +62,7 @@ can run_inference(model_name: str, num_samples: int, payload: dict) -> dict {
                 full_prompt = ret["full_prompt"];
             }
         } except Exception as e  {
-            outputs.append({"response":str(e), "time":-1 });
+            outputs.append({"response":str(e), "time": -1 });
         }
     }
     avg_time = sum([o["time"]  for o in outputs]) / len(outputs);

diff --git a/src/tests/test_generator.jac b/src/tests/test_generator.jac
@@ -46,7 +46,7 @@ test generator   {
     get_item_by_label(generator_tab, "text_input", "thing").set_value("life").run();
     generator_tab = get_item_by_label(app, "tab", "Response Generator");
     assert not generator_tab.error;
-    get_item_by_label(generator_tab, "button", "Generate Responses").set_value(True).run();
+    get_item_by_label(generator_tab, "button", "Generate Responses").set_value(True).run(timeout=20);
     assert os.path.exists(os.path.join("runs", app.session_state.run_id, "orca-mini:3b.json"));
     assert os.path.exists(os.path.join("data", f"{app.session_state.run_id}_responses.json"));
     query_engine.terminate();