Create the subfolder if it doesn't exist when creating an incomplete file

PiperOrigin-RevId: 701935891

Author: tomvdw

tensorflow_datasets/core/utils/py_utils.py

@@ -334,6 +334,7 @@ def incomplete_file(
 ) -> Iterator[epath.Path]:
   """Writes to path atomically, by writing to temp file and renaming it."""
   tmp_path = _tmp_file_name(path, subfolder=subfolder)
+  tmp_path.parent.mkdir(exist_ok=True)
   try:
     yield tmp_path
     tmp_path.replace(path)