Update logging to not change root logger

Signed-off-by: Ayush Dattagupta <[email protected]>
NVIDIA · Apr 23, 2024 · 7bc1536 · 7bc1536
1 parent 5bad027
commit 7bc1536
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/nemo_curator/utils/import_utils.py b/nemo_curator/utils/import_utils.py
@@ -23,8 +23,9 @@
 
 from nemo_curator.utils.gpu_utils import GPU_INSTALL_STRING
 
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger("import_logger")
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO)
+logger.addHandler(logging.StreamHandler())
 
 
 class UnavailableError(Exception):