Add missing method (cvat-ai#120)

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