Add an allow-list when checking versions in all datasets and rollback imagenet_v2's version.

PiperOrigin-RevId: 690541469

Author: marcenacp

tensorflow_datasets/datasets/imagenet_v2/imagenet_v2_dataset_builder.py

@@ -69,7 +69,7 @@ def _create_builder_configs():
 class Builder(tfds.core.GeneratorBasedBuilder):
   """An ImageNet test set recollected by following the original protocol."""
 
-  VERSION = tfds.core.Version('3.1.0')
+  VERSION = tfds.core.Version('3.0.0')
   SUPPORTED_VERSIONS = [
       tfds.core.Version('2.0.0'),
   ]

tensorflow_datasets/testing/version_test.py

@@ -18,6 +18,8 @@
 
 import tensorflow_datasets as tfds
 
+_DO_NOT_CKECK_DATASETS = ['imagenet_v2']
+
 
 @dataclasses.dataclass(frozen=True, kw_only=True)
 class _BuilderWithVersionMismatch:
@@ -33,6 +35,8 @@ def test_internal_datasets_have_versions_on_line_with_the_release_notes():
   builders_with_version_mismatch: list[_BuilderWithVersionMismatch] = []
   for builder in builders:
     builder_cls = tfds.core.registered.imported_builder_cls(builder)
+    if builder_cls.name in _DO_NOT_CKECK_DATASETS:
+      continue
     if not (
         hasattr(builder_cls, 'VERSION')
         and hasattr(builder_cls, 'RELEASE_NOTES')