diff --git a/src/psyc/lines/train.py b/src/psyc/lines/train.py index 0ebadd7..3610c94 100644 --- a/src/psyc/lines/train.py +++ b/src/psyc/lines/train.py @@ -105,7 +105,7 @@ def _ex_severity_classification(case: Case) -> Optional[Example]: "summary": case.summary, "source_type": case.source_type, "incident_type": case.classification.incident_type.value if case.classification.incident_type else None, - "url_status": case.source_metadata.get("url_status", ""), + "status": case.source_metadata.get("url_status") or case.source_metadata.get("status", ""), "victim_country": case.victim.country, "critical_infrastructure": case.victim.critical_infrastructure, "url_count": len(case.observables.urls),