fix: disable logging in argilla

Aleph-Alpha · Jun 17, 2024 · a874938 · a874938
1 parent 26e5fb6
commit a874938
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/intelligence_layer/connectors/argilla/argilla_wrapper_client.py b/src/intelligence_layer/connectors/argilla/argilla_wrapper_client.py
@@ -1,4 +1,5 @@
 import itertools
+import logging
 import os
 from collections.abc import Iterable, Sequence
 from typing import (
@@ -31,6 +32,11 @@ def __init__(
             import warnings
 
             warnings.filterwarnings("ignore", module="argilla.*")
+            # this logger is set on info for some reason
+            logging.getLogger("argilla.client.feedback.dataset.local.mixins").setLevel(
+                logging.WARNING
+            )
+
         rg.init(
             api_url=api_url if api_url is not None else os.getenv("ARGILLA_API_URL"),
             api_key=api_key if api_key is not None else os.getenv("ARGILLA_API_KEY"),