Merge pull request #59 from fastlabel/bugfix/export-yolo-no-annotations

fix yolo export with no annotations

Author: faycute

fastlabel/__init__.py

@@ -578,7 +578,9 @@ def export_yolo(self, tasks: list, output_dir: str = os.path.join("output", "yol
                 for obj in anno["object"]:
                     f.write(obj)
                     f.write("\n")
-        with open(os.path.join(output_dir, "classes.txt"), 'w', encoding="utf8") as f:
+        classes_file_path = os.path.join(output_dir, "classes.txt")
+        os.makedirs(os.path.dirname(classes_file_path), exist_ok=True)
+        with open(classes_file_path, 'w', encoding="utf8") as f:
             for category in categories:
                 f.write(category["name"])
                 f.write("\n")

setup.py

@@ -8,7 +8,7 @@
 
 setuptools.setup(
     name="fastlabel",
-    version="0.9.9",
+    version="0.9.10",
     author="eisuke-ueta",
     author_email="[email protected]",
     description="The official Python SDK for FastLabel API, the Data Platform for AI",