Prepare for changing the Python type checker.

PiperOrigin-RevId: 941653046

python/riegeli/records/tests/records_test.py

diff --git a/python/riegeli/records/tests/records_test.py b/python/riegeli/records/tests/records_test.py
index 4b3640e..bcd3275 100644
--- a/python/riegeli/records/tests/records_test.py
+++ b/python/riegeli/records/tests/records_test.py
@@ -280,7 +280,7 @@
     byte_writer = FakeFile(random_access is RandomAccess.RANDOM_ACCESS)
     with self.assertRaises(NotImplementedError):
       with riegeli.RecordWriter(
-          byte_writer,
+          byte_writer,  # pyrefly: ignore[bad-argument-type]
           assumed_pos=(
               0
               if random_access is RandomAccess.SEQUENTIAL_ACCESS_EXPLICIT
@@ -295,7 +295,7 @@
     byte_reader = FakeFile(random_access is RandomAccess.RANDOM_ACCESS)
     with self.assertRaises(NotImplementedError):
       with riegeli.RecordReader(
-          byte_reader,
+          byte_reader,  # pyrefly: ignore[bad-argument-type]
           owns_src=False,
           assumed_pos=(
               0
@@ -527,7 +527,7 @@
       ) as reader:
         metadata_read = reader.read_metadata()
         self.assertEqual(metadata_read, metadata_written)
-        record_type = riegeli.get_record_type(metadata_read)
+        record_type = riegeli.get_record_type(metadata_read)  # pyrefly: ignore[bad-argument-type]
         assert record_type is not None
         self.assertEqual(
             record_type.DESCRIPTOR.full_name, 'riegeli.tests.SimpleMessage'

python/riegeli/tensorflow/kernel_tests/riegeli_dataset_test.py

diff --git a/python/riegeli/tensorflow/kernel_tests/riegeli_dataset_test.py b/python/riegeli/tensorflow/kernel_tests/riegeli_dataset_test.py
index 54b7a47..02ce0a1 100644
--- a/python/riegeli/tensorflow/kernel_tests/riegeli_dataset_test.py
+++ b/python/riegeli/tensorflow/kernel_tests/riegeli_dataset_test.py
@@ -168,7 +168,7 @@
           dataset, requires_initialization=requires_initialization
       )
       result = []
-      for _ in range(len(expected_output)):
+      for _ in range(len(expected_output)):  # pyrefly: ignore[bad-argument-type]
         result.append(self.evaluate(get_next()))
       self._compareOutputToExpected(result, expected_output, assert_items_equal)
       with self.assertRaises(errors.OutOfRangeError):
@@ -206,7 +206,9 @@
 
       # Note: if records were serialized proto messages, passing
       # options='transpose' to RecordWriter would make compression better.
-      with riegeli.RecordWriter(tf.io.gfile.GFile(filename, 'wb')) as writer:
+      with riegeli.RecordWriter(
+          tf.io.gfile.GFile(filename, 'wb')  # pyrefly: ignore[bad-argument-type]
+      ) as writer:
         for j in range(self._num_records):
           writer.write_record(self._record(i, j))
     return filenames