Fix project export (cvat-ai#119)

TOsmanov · Feb 26, 2021 · 9a88cad · 9a88cad
1 parent 68b1212
commit 9a88cad
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/datumaro/components/project.py b/datumaro/components/project.py
@@ -277,6 +277,10 @@ def categories(_):
 
         return self.transform(_DatasetFilter)
 
+    def export(self, save_dir: str, format, **kwargs):
+        dataset = Dataset.from_extractors(self, env=self.env)
+        dataset.export(save_dir, format, **kwargs)
+
     def transform_project(self, method, save_dir=None, **method_kwargs):
         # NOTE: probably this function should be in the ViewModel layer
         if isinstance(method, str):