Fix Failing Unittests introduced in #650 (#651)

* update new warning format with rank Signed-off-by: Jason <[email protected]> * add explicit rank marker Signed-off-by: Jason <[email protected]>
NVIDIA · May 21, 2020 · 4951396 · 4951396
1 parent 8a7c94f
commit 4951396
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nemo/utils/formatters/base.py b/nemo/utils/formatters/base.py
@@ -130,5 +130,5 @@ class BaseNeMoFormatter(BaseFormatter):
 
 class DebugNeMoFormatter(BaseFormatter):
     DEFAULT_FORMAT = (
-        "%(color)s[NeMo %(levelname)1.1s %(asctime)s %(module)s:%(lineno)d %(rank)d]%(end_color)s %(message)s"
+        "%(color)s[NeMo %(levelname)1.1s %(asctime)s %(module)s:%(lineno)d rank:%(rank)d]%(end_color)s %(message)s"
     )
diff --git a/tests/unit/utils/test_deprecated.py b/tests/unit/utils/test_deprecated.py
@@ -30,7 +30,7 @@
 
 class DeprecatedTest(TestCase):
     NEMO_ERR_MSG_FORMAT = re.compile(
-        r"\[NeMo W [0-9]{4}-[0-9]{2}-[0-9]{2} [0-9]{2}:[0-9]{2}:[0-9]{2} deprecated:[0-9]*\] "
+        r"\[NeMo W [0-9]{4}-[0-9]{2}-[0-9]{2} [0-9]{2}:[0-9]{2}:[0-9]{2} deprecated:[0-9]+( rank:[0-9]+)?\] "
     )
 
     @pytest.mark.unit