Assert match is not None

databrickslabs · Apr 4, 2024 · 8d9f12a · 8d9f12a
1 parent 35cd5c2
commit 8d9f12a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/databricks/labs/ucx/installer/logs.py b/src/databricks/labs/ucx/installer/logs.py
@@ -57,6 +57,7 @@ def parse_logs(*log_paths: Path) -> Iterator[LogRecord]:
                     next_line = f.readline()
                     next_match = pattern.match(next_line)
 
+                assert match is not None
                 level, msg = match.groups()
                 log_record = LogRecord(logging.getLevelName(level), msg + multi_line_message)
                 yield log_record

diff --git a/tests/unit/installer/test_logs.py b/tests/unit/installer/test_logs.py
@@ -25,7 +25,7 @@
     "07:09 INFO [module] Other message.\n",
     "07:09 WARNING [module] Warning message.\n",
     "07:09 CRITICAL [module] Watch out!\n",
-    f"04:36 DEBUG [databricks.sdk] {MULTILINE_LOG_MESSAGE}"
+    f"04:36 DEBUG [databricks.sdk] {MULTILINE_LOG_MESSAGE}",
 ]
 LOG_RECORDS = [
     LogRecord(40, "Message."),