Fix lint

stanford-crfm · Feb 3, 2025 · 779ba1c · 779ba1c
1 parent 2bca0e9
commit 779ba1c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts/redact_scenario_states.py b/scripts/redact_scenario_states.py
@@ -77,6 +77,7 @@ def redact_result(result: RequestResult) -> RequestResult:
 
 
 def redact_request_state(request_state: RequestState, redact_output: bool) -> RequestState:
+    assert request_state.result is not None
     result = redact_result(request_state.result) if redact_output else request_state.result
     return dataclasses.replace(
         request_state,