refactor: Enhance logging configuration in W&B to Neptune migration s…

…cript
neptune-ai · Jan 9, 2025 · 5966f41 · 5966f41
1 parent 9b84878
commit 5966f41
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/utils/migration_tools/from_wandb/wandb_to_neptune.py b/utils/migration_tools/from_wandb/wandb_to_neptune.py
@@ -32,6 +32,12 @@
 from neptune_scale.projects import create_project
 from tqdm.auto import tqdm
 
+# Silencing Neptune and httpx messages, W&B errors, urllib connection pool warnings
+logging.getLogger("httpx").setLevel(logging.ERROR)
+logging.getLogger("urllib3").setLevel(logging.ERROR)
+logging.getLogger("wandb").setLevel(logging.ERROR)
+logging.getLogger("neptune").setLevel(logging.CRITICAL)
+
 SUPPORTED_DATATYPES = [int, float, str, datetime, bool, list, set]
 
 if __name__ == "__main__":
@@ -84,12 +90,6 @@
 
     print(f"Logs available at {log_filename}\n")
 
-    # Silencing Neptune and httpx messages, W&B errors, urllib connection pool warnings
-    logging.getLogger("neptune").setLevel(logging.CRITICAL)
-    logging.getLogger("httpx").setLevel(logging.ERROR)
-    logging.getLogger("urllib3").setLevel(logging.ERROR)
-    logging.getLogger("wandb").setLevel(logging.ERROR)
-
     def exc_handler(exctype, value, tb):
         logger.exception("".join(traceback.format_exception(exctype, value, tb)))