fix: set connection autocommit to True

huggingface · Dec 18, 2024 · c2d4b82 · c2d4b82
1 parent f990835
commit c2d4b82
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/benchmark/benchmarks_entrypoint.py b/benchmark/benchmarks_entrypoint.py
@@ -20,6 +20,7 @@ class ImportModuleException(Exception):
 class MetricsRecorder:
     def __init__(self, connection, logger: logging.Logger, branch: str, commit_id: str, commit_msg: str):
         self.conn = connection
+        self.conn.autocommit = True
         self.logger = logger
         self.branch = branch
         self.commit_id = commit_id
@@ -73,10 +74,10 @@ def close(self):
 
 
 logger = logging.getLogger(__name__)
-logger.setLevel(logging.DEBUG)
+logger.setLevel(logging.INFO)
 
 handler = logging.StreamHandler(sys.stdout)
-handler.setLevel(logging.DEBUG)
+handler.setLevel(logging.INFO)
 formatter = logging.Formatter("[%(levelname)s - %(asctime)s] %(message)s")
 handler.setFormatter(formatter)
 logger.addHandler(handler)