Suppress new Pyrefly findings before upgrading to v1.2.0

PiperOrigin-RevId: 979034554

Author: oprypin

tensorflow_datasets/core/community/dataset_sources.py

@@ -57,7 +57,7 @@ def to_json(self) -> utils.JsonValue:
     else:
       return {  # pyrefly: ignore[bad-return]
           'root_path': os.fspath(self.root_path),
-          'filenames': self.filenames,
+          'filenames': self.filenames,  # pyrefly: ignore[bad-assignment]
       }
 
 

tensorflow_datasets/core/dataset_builders/croissant_builder.py

@@ -430,7 +430,7 @@ def _split_generators(
           for split in split_reference.split_record_set.data  # pyrefly: ignore[not-iterable]
       }
     else:
-      return {'default': self._generate_examples(filters=self._filters)}  # pyrefly: ignore[bad-argument-type, bad-return]
+      return {'default': self._generate_examples(filters=self._filters)}  # pyrefly: ignore[bad-argument-type, bad-assignment, bad-return]
 
   def _generate_examples(
       self,

tensorflow_datasets/core/folder_dataset/compute_split_utils.py

@@ -233,7 +233,7 @@ def _compute_split_statistics(
           functools.partial(
               _process_shard,
               data_dir=filename_template.data_dir,
-              adapter=adapter,
+              adapter=adapter,  # pyrefly: ignore[bad-argument-type]
           ),
           split_files,
           progress_bar=True,

tensorflow_datasets/core/reader.py

@@ -484,7 +484,7 @@ def parse_and_decode(ex: Tensor) -> TreeDict[Tensor]:
     # Eventually add the `tfds_id` after the decoding
     if read_config and read_config.add_tfds_id:
       parse_and_decode = functools.partial(
-          _decode_with_id, decode_fn=parse_and_decode
+          _decode_with_id, decode_fn=parse_and_decode  # pyrefly: ignore[bad-argument-type]
       )
 
     ds = ds.map(

tensorflow_datasets/object_detection/waymo_open_dataset.py

@@ -290,8 +290,8 @@ def _generate_images_and_annotations(tf_record_file):
 
       camera_name = open_dataset.CameraName.Name.Name(frame_image.name)
       image_and_annotation["camera_" + camera_name] = {  # pyrefly: ignore[unsupported-operation]
-          "image": frame_image.image,
-          "labels": labels,
+          "image": frame_image.image,  # pyrefly: ignore[bad-assignment]
+          "labels": labels,  # pyrefly: ignore[bad-assignment]
       }
 
     yield image_and_annotation

tensorflow_datasets/rl_unplugged/rlu_locomotion/rlu_locomotion.py

@@ -173,7 +173,7 @@ def get_features_dict(self):
     return tfds.features.FeaturesDict({
         'steps': tfds.features.Dataset({  # pyrefly: ignore[bad-argument-type]
             'observation': {
-                'walker': walker_features,
+                'walker': walker_features,  # pyrefly: ignore[bad-assignment]
             },
             'action': action_features,
             'reward': np.float32,

tensorflow_datasets/robomimic/dataset_utils.py

@@ -303,7 +303,7 @@ def _get_features(self) -> tfds.features.FeaturesDict:
         'episode_id': np.str_,
         'steps': tfds.features.Dataset({  # pyrefly: ignore[bad-argument-type]
             'action': tensor_feature(action_size),  # pyrefly: ignore[bad-argument-type]
-            'observation': observation,
+            'observation': observation,  # pyrefly: ignore[bad-assignment]
             'reward': np.float64,
             'is_first': np.bool_,
             'is_last': np.bool_,

tensorflow_datasets/structured/wine_quality/wine_quality.py

@@ -121,7 +121,7 @@ def _info(self):
         description=_DESCRIPTION,
         features=tfds.features.FeaturesDict({  # pyrefly: ignore[bad-argument-type]
             "quality": np.int32,
-            "features": features_dict,
+            "features": features_dict,  # pyrefly: ignore[bad-assignment]
         }),
         supervised_keys=("features", "quality"),
         homepage=_HOMEPAGE_URL,