fix/non_folder_export

databricks · Jan 5, 2024 · 2b07e69 · 2b07e69
1 parent 71ce20a
commit 2b07e69
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lilac/router_dataset.py b/lilac/router_dataset.py
@@ -294,7 +294,10 @@ def serve_dataset_file(filepath: str) -> FileResponse:
 def export_dataset(namespace: str, dataset_name: str, options: ExportOptions) -> str:
   """Export the dataset to one of the supported file formats."""
   dataset = get_dataset(namespace, dataset_name)
-  os.makedirs(os.path.dirname(options.filepath), exist_ok=True)
+  # Only create the directory if the filepath has a directory.
+  dirname = os.path.dirname(options.filepath)
+  if dirname:
+    os.makedirs(dirname, exist_ok=True)
 
   if options.format == 'csv':
     dataset.to_csv(