From bdf124a550e34db2cc56b135bc97d2a73e685253 Mon Sep 17 00:00:00 2001 From: Dmitrii Lavrukhin Date: Wed, 23 Oct 2024 19:38:00 +0400 Subject: [PATCH] fix missed imports --- tests/unit/test_coco_format.py | 1 + tests/unit/test_tfrecord_format.py | 1 + 2 files changed, 2 insertions(+) diff --git a/tests/unit/test_coco_format.py b/tests/unit/test_coco_format.py index 3363510378..dcd18ba9a3 100644 --- a/tests/unit/test_coco_format.py +++ b/tests/unit/test_coco_format.py @@ -48,6 +48,7 @@ from datumaro.util import dump_json_file from tests.requirements import Requirements, mark_requirement +from tests.utils.assets import get_test_asset_path from tests.utils.test_utils import ( TestDir, check_save_and_load, diff --git a/tests/unit/test_tfrecord_format.py b/tests/unit/test_tfrecord_format.py index 7bd430552c..8f69f5bed0 100644 --- a/tests/unit/test_tfrecord_format.py +++ b/tests/unit/test_tfrecord_format.py @@ -14,6 +14,7 @@ from datumaro.util.tf_util import check_import from tests.requirements import Requirements, mark_requirement +from tests.utils.assets import get_test_asset_path from tests.utils.test_utils import TestDir, check_save_and_load, compare_datasets try: