gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-02 09:57:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: handle invalid regex patterns in filter_progress_bar function (#579)

Browse Source 
* fix: handle invalid regex patterns in filter_progress_bar function

* Update rdagent/utils/__init__.py
This commit is contained in:
you-n-g
2025-02-11 18:07:16 +08:00
committed by GitHub
parent 75f29f0dd9
commit 56f3a668f2
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/utils/__init__.py
+12 -8
View File
@@ -123,16 +123,20 @@ def filter_progress_bar(stdout: str) -> str:
)
needs_sub = response.get("needs_sub", True)
regex_patterns = response.get("regex_patterns", [])
if isinstance(regex_patterns, list):
for pattern in regex_patterns:
filtered_stdout = re.sub(pattern, "", filtered_stdout)
else:
filtered_stdout = re.sub(regex_patterns, "", filtered_stdout)
try:
if isinstance(regex_patterns, list):
for pattern in regex_patterns:
filtered_stdout = re.sub(pattern, "", filtered_stdout)
else:
filtered_stdout = re.sub(regex_patterns, "", filtered_stdout)
if not needs_sub:
break
filtered_stdout = re.sub(r"\s*\n\s*", "\n", filtered_stdout)
if not needs_sub:
break
filtered_stdout = re.sub(r"\s*\n\s*", "\n", filtered_stdout)
except re.error as e: # sometime the generated regex pattern is invalid and yield exception.
from rdagent.log import rdagent_logger as logger
logger.error(f"Error in filtering progress bar: due to {e}")
return filtered_stdout