Logging

Signed-off-by: Tyler Gu <[email protected]>
xlab-uiuc · Sep 6, 2024 · 9238384 · 9238384
1 parent 0bb67e6
commit 9238384
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/acto/schema/array.py b/acto/schema/array.py
@@ -1,7 +1,8 @@
-import logging
 import random
 from typing import List, Tuple
 
+from utils.thread_logger import get_thread_logger
+
 from .base import BaseSchema, TreeNode
 
 
@@ -102,7 +103,8 @@ def to_tree(self) -> TreeNode:
         return node
 
     def load_examples(self, example: list):
-        logging.debug(f"Loading example {example} into {self}")
+        logger = get_thread_logger(with_prefix=True)
+        logger.debug(f"Loading example {example} into {self}")
         self.examples.append(example)
         for item in example:
             self.item_schema.load_examples(item)

diff --git a/acto/schema/object.py b/acto/schema/object.py
@@ -1,4 +1,3 @@
-import logging
 import random
 from typing import List, Tuple
 
@@ -130,7 +129,8 @@ def to_tree(self) -> TreeNode:
         return node
 
     def load_examples(self, example: dict):
-        logging.debug(f"Loading example {example} into {self}")
+        logger = get_thread_logger(with_prefix=True)
+        logger.debug(f"Loading example {example} into {self}")
         self.examples.append(example)
         for key, value in example.items():
             if key in self.properties:

diff --git a/acto/schema/string.py b/acto/schema/string.py
@@ -1,8 +1,8 @@
-import logging
 import random
 from typing import List, Optional, Tuple
 
 import exrex
+from utils.thread_logger import get_thread_logger
 
 from acto.common import random_string
 
@@ -48,7 +48,8 @@ def to_tree(self) -> TreeNode:
         return TreeNode(self.path)
 
     def load_examples(self, example: str):
-        logging.debug(f"Loading example {example} into {self}")
+        logger = get_thread_logger(with_prefix=True)
+        logger.debug(f"Loading example {example} into {self}")
         self.examples.append(example)
 
     def set_default(self, instance):