fix a typing annotation

PiperOrigin-RevId: 716682376
tensorflow · Jan 17, 2025 · 3fe07bd · 3fe07bd
1 parent 4baec15
commit 3fe07bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_datasets/testing/dataset_collection_builder_testing.py b/tensorflow_datasets/testing/dataset_collection_builder_testing.py
@@ -49,7 +49,7 @@ class DatasetCollectionTestBase:
     the collection will be checked.
   """
 
-  DATASET_COLLECTION_CLASS = None
+  DATASET_COLLECTION_CLASS: type[dataset_collection_builder.DatasetCollection]
   VERSION: Optional[str] = None
   DATASETS_TO_TEST: List[str] = []
   CHECK_DATASETS_VERSION: bool = True