Merge remote-tracking branch 'origin/main' into pre_pr

This commit is contained in:
Ori Kotek
2023-08-30 15:53:59 +03:00
11 changed files with 412 additions and 127 deletions

View File

@ -286,7 +286,7 @@ def _fix_key_value(key: str, value: str):
def load_yaml(review_text: str) -> dict:
review_text = review_text.removeprefix('```yaml').rstrip('`')
try:
data = yaml.load(review_text, Loader=yaml.SafeLoader)
data = yaml.safe_load(review_text)
except Exception as e:
logging.error(f"Failed to parse AI prediction: {e}")
data = try_fix_yaml(review_text)