Use the file suffix when updating split

tomvdw · The TensorFlow Datasets Authors · commit d69632d9cada · 2024-07-16T03:47:45.000-07:00
Currently it incorrectly sets it to the file format enum value, which may or may not correspond to the file suffix.

PiperOrigin-RevId: 652773225
diff --git a/tensorflow_datasets/core/dataset_info.py b/tensorflow_datasets/core/dataset_info.py
@@ -512,7 +512,7 @@ def set_file_format(
           continue
         updated_split_info = split_info.replace(
             filename_template=split_info.filename_template.replace(
-                filetype_suffix=file_format.value
+                filetype_suffix=file_format.file_suffix
             )
         )
         updated_split_infos.append(updated_split_info)

Original file line number	Diff line number	Diff line change
`@@ -512,7 +512,7 @@ def set_file_format(`
`512`	`512`	`continue`
`513`	`513`	`updated_split_info = split_info.replace(`
`514`	`514`	`filename_template=split_info.filename_template.replace(`
`515`		`- filetype_suffix=file_format.value`
	`515`	`+ filetype_suffix=file_format.file_suffix`
`516`	`516`	`)`
`517`	`517`	`)`
`518`	`518`	`updated_split_infos.append(updated_split_info)`