fix test

cvat-ai · Aug 3, 2020 · 0974d88 · 0974d88
1 parent bef0927
commit 0974d88
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/datumaro/tests/test_ops.py b/datumaro/tests/test_ops.py
@@ -1,7 +1,8 @@
 import numpy as np
 
 from datumaro.components.extractor import (Extractor, DatasetItem, Label,
- Mask, Bbox, Points, LabelCategories, AnnotationType, Caption)
+ Mask, Bbox, Points, Caption)
+from datumaro.components.project import Dataset
 from datumaro.components.operations import mean_std, compute_ann_statistics
 
 from unittest import TestCase
@@ -128,6 +129,6 @@ def test_stats(self):
  },
  }
 
- actual = compute_ann_statistics(TestExtractor())
+ actual = compute_ann_statistics(dataset)
 
  self.assertEqual(expected, actual)