Merge pull request #822 from samskillman/patch-2

Fix double/single colon regex for py-nemo
GoogleCloudPlatform · Jan 14, 2025 · 977afbc · 977afbc
2 parents c257d68 + e0ee3ae
commit 977afbc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/var/ramble/repos/builtin/applications/py-nemo/application.py b/var/ramble/repos/builtin/applications/py-nemo/application.py
@@ -381,7 +381,7 @@ class PyNemo(ExecutableApplication):
     )
 
     per_epoch_regex = (
-        r"Epoch (?P<epoch_id>[0-9]+):\s+:\s+(?P<pct_complete>[0-9]+)%.*\s+"
+        r"Epoch (?P<epoch_id>[0-9]+)(:\s+)+(?P<pct_complete>[0-9]+)%.*\s+"
         + r"(?P<step_idx>[0-9]+)/(?P<max_itr>[0-9]+) \[(?P<elapsed_time>[0-9:]+)<"
         + r"(?P<remaining_time>[0-9:]+).*"
     )