Revert unrelated pyright fixes in adapter tests.

Keep the branch focused on Runloop rollout changes only.

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: tode-rl

tests/adapters/test_fireworks_tracing_logprobs.py

@@ -71,7 +71,6 @@ def test_attaches_completion_logprobs_and_message_logprobs(self):
 
         assistant = row.messages[-1]
         assert assistant.role == "assistant"
-        assert assistant.logprobs is not None
         content = assistant.logprobs["content"]
         assert len(content) == len(extra["completion_logprobs"])
         assert content[0]["token_id"] == 10
@@ -84,13 +83,10 @@ def test_omits_token_id_keys_when_any_missing(self):
         assert row is not None
 
         extra = row.execution_metadata.extra
-        assert extra is not None
         assert "completion_logprobs" in extra
         assert "completion_token_ids" not in extra
 
-        logprobs = row.messages[-1].logprobs
-        assert logprobs is not None
-        content = logprobs["content"]
+        content = row.messages[-1].logprobs["content"]
         assert len(content) == 2
         assert all("token_id" not in entry for entry in content)
         assert content[0]["logprob"] == pytest.approx(-0.1)

tests/adapters/test_r3_deserializer.py

@@ -143,9 +143,8 @@ def test_all_mode_uint8(self):
         assert metadata["replayed_token_count"] == total_tokens
 
         for i in range(total_tokens):
-            matrix = matrices[i]
-            assert matrix is not None
-            decoded = base64.b64decode(matrix)
+            assert matrices[i] is not None
+            decoded = base64.b64decode(matrices[i])
             assert decoded == matrices_raw[i]
 
     def test_suffix_mode(self):
@@ -181,9 +180,8 @@ def test_suffix_mode(self):
         # Positions from start_token to start_token+replayed should have data
         for i in range(replayed):
             pos = start_token + i
-            matrix = matrices[pos]
-            assert matrix is not None
-            decoded = base64.b64decode(matrix)
+            assert matrices[pos] is not None
+            decoded = base64.b64decode(matrices[pos])
             assert decoded == matrices_raw[i]
 
     def test_bitmap_mode(self):
@@ -222,10 +220,9 @@ def test_bitmap_mode(self):
 
         for i in range(total_tokens):
             if i in replayed_positions:
-                matrix = matrices[i]
-                assert matrix is not None
+                assert matrices[i] is not None
                 idx = replayed_positions.index(i)
-                decoded = base64.b64decode(matrix)
+                decoded = base64.b64decode(matrices[i])
                 assert decoded == matrices_raw[idx]
             else:
                 assert matrices[i] is None
@@ -252,9 +249,7 @@ def test_uint16_dtype(self):
         assert metadata["routing_dtype"] == "uint16"
         assert len(matrices) == total_tokens
         for i in range(total_tokens):
-            matrix = matrices[i]
-            assert matrix is not None
-            decoded = base64.b64decode(matrix)
+            decoded = base64.b64decode(matrices[i])
             assert decoded == matrices_raw[i]
 
     def test_zero_replayed_tokens(self):
@@ -307,7 +302,6 @@ def test_high_compression_ratio_payload(self):
         assert len(matrices) == total_tokens
         assert metadata["replayed_token_count"] == total_tokens
         for m in matrices:
-            assert m is not None
             assert base64.b64decode(m) == b"\x00" * matrix_elem_size