Fix flake8 in demo paper

Author: sergioburdisso

tutorials/demo_paper/2.dialog_generation.py

@@ -24,40 +24,38 @@
     rules="Ask general questions about services"
 )
 
+
 def generate_customers(base_customer, n, save_folder):
 
-  cgen = PersonaGenerator(base_customer)
-  cgen.set(
-    politeness=["rude", "neutral", "high"]
-  )
+    cgen = PersonaGenerator(base_customer)
+    cgen.set(
+        politeness=["rude", "neutral", "high"]
+    )
 
-  customers = []
-  for ix in tqdm(range(n), desc="Generating customers"):
-    path = os.path.join(save_folder, f"customer_{ix}.json")
+    customers = []
+    for ix in tqdm(range(n), desc="Generating customers"):
+        path = os.path.join(save_folder, f"customer_{ix}.json")
 
-    if not os.path.exists(path):
-        customer = cgen.generate()  # Generate a new customer persona!
-        customer.to_file(path)
-    else:
-        customer = Customer.from_file(path)
-    customers.append(customer)
+        if not os.path.exists(path):
+            customer = cgen.generate()  # Generate a new customer persona!
+            customer.to_file(path)
+        else:
+            customer = Customer.from_file(path)
+        customers.append(customer)
 
-  return customers
+    return customers
 
 
 def generate_dialogs(llm_name, customer, n, save_folder):
 
-  agent = build_my_agent(llm_name)
+    agent = build_my_agent(llm_name)
 
-  customer = Agent(
-    persona=customer,
-    name="Customer"
-  )
+    customer = Agent(persona=customer, name="Customer")
 
-  for ix in tqdm(range(n), desc="Generating dialogs"):
-    if not os.path.exists(os.path.join(save_folder, f"dialog_{ix}.json")):
-        dialog = agent.talk_with(customer)
-        dialog.to_file(os.path.join(save_folder, f"dialog_{ix}.json"))
+    for ix in tqdm(range(n), desc="Generating dialogs"):
+        if not os.path.exists(os.path.join(save_folder, f"dialog_{ix}.json")):
+            dialog = agent.talk_with(customer)
+            dialog.to_file(os.path.join(save_folder, f"dialog_{ix}.json"))
 
 
 # Case A: requiring verification
@@ -68,9 +66,9 @@ def generate_dialogs(llm_name, customer, n, save_folder):
                                     "output/no_verification/customers")
 
 for llm in tqdm(LLMS, desc="Processing LLMs"):
-  # Case A: requiring verification
-  for customer in tqdm(customers_v, desc=f"Customers (verification) - {llm}"):
-    generate_dialogs(llm, customer, NUM_DIALOGS, f"output/requires_verification/{llm}/")
-  # Case B: not requiring verification
-  for customer in tqdm(customers_no_v, desc=f"Customers (no verification) - {llm}"):
-    generate_dialogs(llm, customer, NUM_DIALOGS, f"output/no_verification/{llm}/")
+    # Case A: requiring verification
+    for customer in tqdm(customers_v, desc=f"Customers (verification) - {llm}"):
+        generate_dialogs(llm, customer, NUM_DIALOGS, f"output/requires_verification/{llm}/")
+    # Case B: not requiring verification
+    for customer in tqdm(customers_no_v, desc=f"Customers (no verification) - {llm}"):
+        generate_dialogs(llm, customer, NUM_DIALOGS, f"output/no_verification/{llm}/")

tutorials/demo_paper/3.evaluation.py

@@ -48,20 +48,20 @@
 comparator_v = Comparator(evaluators=[freq_judge_ask_v, freq_tool_seq_v])
 print("\nResults - Requires Verification")
 comparator_v({
-  "qwen3:0.6b": Dialog.from_folder("output/requires_verification/qwen3:0.6b/"),
-  "qwen3:1.7b": Dialog.from_folder("output/requires_verification/qwen3:1.7b/"),
-  "qwen3:8b": Dialog.from_folder("output/requires_verification/qwen3:8b/"),
-  "qwen3:14b": Dialog.from_folder("output/requires_verification/qwen3:14b/")
+    "qwen3:0.6b": Dialog.from_folder("output/requires_verification/qwen3:0.6b/"),
+    "qwen3:1.7b": Dialog.from_folder("output/requires_verification/qwen3:1.7b/"),
+    "qwen3:8b": Dialog.from_folder("output/requires_verification/qwen3:8b/"),
+    "qwen3:14b": Dialog.from_folder("output/requires_verification/qwen3:14b/")
 })
 comparator_v.plot(save_folder_path="output/requires_verification")
 
 # Case B: not requiring verification
 comparator_no_v = Comparator(evaluators=[freq_judge_ask_v, freq_tool_seq_no_v])
 print("\nResults - No Verification Required")
 comparator_no_v({
-  "qwen3:0.6b": Dialog.from_folder("output/no_verification/qwen3:0.6b/"),
-  "qwen3:1.7b": Dialog.from_folder("output/no_verification/qwen3:1.7b/"),
-  "qwen3:8b": Dialog.from_folder("output/no_verification/qwen3:8b/"),
-  "qwen3:14b": Dialog.from_folder("output/no_verification/qwen3:14b/")
+    "qwen3:0.6b": Dialog.from_folder("output/no_verification/qwen3:0.6b/"),
+    "qwen3:1.7b": Dialog.from_folder("output/no_verification/qwen3:1.7b/"),
+    "qwen3:8b": Dialog.from_folder("output/no_verification/qwen3:8b/"),
+    "qwen3:14b": Dialog.from_folder("output/no_verification/qwen3:14b/")
 })
 comparator_no_v.plot(save_folder_path="output/no_verification")