import and bug fixes

Signed-off-by: Jason <[email protected]>

nemo/utils/formatters/base.py

@@ -115,5 +115,5 @@ def format(self, record):
         return formatted.replace("\n", "\n    ")
 
 
-class BaseNeMoFormatter(logging.Formatter):
+class BaseNeMoFormatter(BaseFormatter):
     DEFAULT_FORMAT = "%(color)s[NeMo %(levelname)1.1s %(asctime)s %(module)s:%(lineno)d]%(end_color)s %(message)s"

nemo/utils/formatters/utils.py

@@ -3,7 +3,7 @@
 
 from nemo.constants import NEMO_ENV_VARNAME_ENABLE_COLORING
 
-from nemo.utils import get_envbool
+from nemo.utils.env_var_parsing import get_envbool
 
 __all__ = ["check_color_support", "to_unicode"]
 

nemo/utils/nemo_logging.py

@@ -15,26 +15,26 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
+# import os
 import sys
 
-import inspect
+# import inspect
 import warnings
 
 from contextlib import contextmanager
 
 import threading
 import logging as _logging
 
-from nemo.constants import NEMO_ENV_VARNAME_SAVE_LOGS_TO_DIR
+# from nemo.constants import NEMO_ENV_VARNAME_SAVE_LOGS_TO_DIR
 from nemo.constants import NEMO_ENV_VARNAME_REDIRECT_LOGS_TO_STDERR
 
-from nemo.utils.formatters import BaseNeMoFormatter
+from nemo.utils.formatters.base import BaseNeMoFormatter
 from nemo.utils.metaclasses import SingletonMetaClass
 
-from nemo.utils import get_envbool
-from nemo.utils import get_env
-from nemo.utils import get_envint
+from nemo.utils.env_var_parsing import get_envbool
+from nemo.utils.env_var_parsing import get_env
+from nemo.utils.env_var_parsing import get_envint
 
 __all__ = [
     "Logger",