add split to dataset reference

PiperOrigin-RevId: 642547838

Author: The TensorFlow Datasets Authors

tensorflow_datasets/core/proto/dataset_info.proto

@@ -154,6 +154,12 @@ message TfdsDatasetReference {
   // The namespace of the dataset, e.g. huggingface. Note that we prefix by ds_
   // to avoid conflicts with reserved in keywords in languages like C++.
   string ds_namespace = 5;
+
+  // The split of the dataset. Since the dataset reference is used to refer to
+  // the dataset as a whole, this is usually an empty string, but it can be
+  // set to a specific split if the dataset reference is used to refer to a
+  // specific split.
+  string split = 6;
 }
 
 // This is a serialization of tensorflow_datasets.core.DatasetInfo.

tensorflow_datasets/core/proto/dataset_info_generated_pb2.py

@@ -61,11 +61,12 @@
     b' \x01(\t"$\n\x03Url\x12\x0b\n\x03url\x18\x01'
     b' \x01(\t\x12\x10\n\x08\x63hecksum\x18\x02'
     b' \x01(\t"\x1d\n\x08SqlQuery\x12\x11\n\tsql_query\x18\x01'
-    b' \x01(\t"m\n\x14TfdsDatasetReference\x12\x0c\n\x04name\x18\x01'
+    b' \x01(\t"|\n\x14TfdsDatasetReference\x12\x0c\n\x04name\x18\x01'
     b' \x01(\t\x12\x0e\n\x06\x63onfig\x18\x02'
     b' \x01(\t\x12\x0f\n\x07version\x18\x03'
     b' \x01(\t\x12\x10\n\x08\x64\x61ta_dir\x18\x04'
     b' \x01(\t\x12\x14\n\x0c\x64s_namespace\x18\x05'
+    b' \x01(\t\x12\r\n\x05split\x18\x06'
     b' \x01(\t"\xb4\x07\n\x0b\x44\x61tasetInfo\x12\x0c\n\x04name\x18\x01'
     b' \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02'
     b' \x01(\t\x12\x0f\n\x07version\x18\t \x01(\t\x12I\n\rrelease_notes\x18\x12'
@@ -142,11 +143,11 @@
   _SQLQUERY._serialized_start = 1249
   _SQLQUERY._serialized_end = 1278
   _TFDSDATASETREFERENCE._serialized_start = 1280
-  _TFDSDATASETREFERENCE._serialized_end = 1389
-  _DATASETINFO._serialized_start = 1392
-  _DATASETINFO._serialized_end = 2340
-  _DATASETINFO_RELEASENOTESENTRY._serialized_start = 2231
-  _DATASETINFO_RELEASENOTESENTRY._serialized_end = 2282
-  _DATASETINFO_DOWNLOADCHECKSUMSENTRY._serialized_start = 2284
-  _DATASETINFO_DOWNLOADCHECKSUMSENTRY._serialized_end = 2340
+  _TFDSDATASETREFERENCE._serialized_end = 1404
+  _DATASETINFO._serialized_start = 1407
+  _DATASETINFO._serialized_end = 2355
+  _DATASETINFO_RELEASENOTESENTRY._serialized_start = 2246
+  _DATASETINFO_RELEASENOTESENTRY._serialized_end = 2297
+  _DATASETINFO_DOWNLOADCHECKSUMSENTRY._serialized_start = 2299
+  _DATASETINFO_DOWNLOADCHECKSUMSENTRY._serialized_end = 2355
 # @@protoc_insertion_point(module_scope)