feat: suppress warning

instadeepai · Apr 21, 2024 · 4a5786e · 4a5786e
1 parent a87ee3d
commit 4a5786e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/debatellm/agents.py b/debatellm/agents.py
@@ -17,7 +17,10 @@
 import time
 from importlib import import_module
 from typing import Any, Callable, Dict, Optional, Tuple
+import logging
 
+# Set the logging level for 'httpx' to 'WARNING' to suppress info and debug messages
+logging.getLogger('httpx').setLevel(logging.WARNING)
 import google
 import numpy as np
 import openai

diff --git a/scripts/experiments_utils.py b/scripts/experiments_utils.py
@@ -141,7 +141,7 @@ def run_experiments(
 
     if verbose:
         print(f"Launching {len(experiments)} experiments...")
-
+        
     with ThreadPoolExecutor(max_workers=parallel_workers) as executor:
         list(tqdm(executor.map(run_experiment, experiments), total=len(experiments)))
 

diff --git a/scripts/launch_experiments.py b/scripts/launch_experiments.py
@@ -211,4 +211,5 @@
         "pubmedqa",
         "mmlu",
     ]  # "medqa", "pubmedqa", "mmlu", "cosmosqa", "ciar", "gpqa", medmcqa
+
 run_experiments(exp_table, parallel_workers=2, shuffle=False, sort_by_dataset=False)