diff --git a/raillabel/__init__.py b/raillabel/__init__.py index 3991e17..3ea3568 100644 --- a/raillabel/__init__.py +++ b/raillabel/__init__.py @@ -3,7 +3,7 @@ """Devkit for working with recorded and annotated train ride data from DB.""" from importlib import metadata -from . import _util, format +from . import format from .exceptions import * from .filter.filter import filter from .format import Scene diff --git a/raillabel/_util/__init__.py b/raillabel/_util/__init__.py deleted file mode 100644 index dd5d085..0000000 --- a/raillabel/_util/__init__.py +++ /dev/null @@ -1,2 +0,0 @@ -# Copyright DB InfraGO AG and contributors -# SPDX-License-Identifier: Apache-2.0 diff --git a/raillabel/_util/_attribute_type.py b/raillabel/format/raillabel/_attribute_type.py similarity index 100% rename from raillabel/_util/_attribute_type.py rename to raillabel/format/raillabel/_attribute_type.py diff --git a/raillabel/format/raillabel/_object_annotation.py b/raillabel/format/raillabel/_object_annotation.py index 941ebdd..6bd484b 100644 --- a/raillabel/format/raillabel/_object_annotation.py +++ b/raillabel/format/raillabel/_object_annotation.py @@ -9,7 +9,7 @@ from pathlib import Path from pkgutil import iter_modules -from ..._util._attribute_type import AttributeType +from ._attribute_type import AttributeType from .object import Object from .sensor import Sensor diff --git a/raillabel/format/raillabel/element_data_pointer.py b/raillabel/format/raillabel/element_data_pointer.py index f0e03cd..0d30a00 100644 --- a/raillabel/format/raillabel/element_data_pointer.py +++ b/raillabel/format/raillabel/element_data_pointer.py @@ -4,7 +4,7 @@ import typing as t from dataclasses import dataclass -from ..._util._attribute_type import AttributeType +from ._attribute_type import AttributeType from .frame_interval import FrameInterval diff --git a/tests/test_raillabel/format/raillabel/test_element_data_pointer.py b/tests/test_raillabel/format/raillabel/test_element_data_pointer.py index 5ee07a4..d3f0eda 100644 --- a/tests/test_raillabel/format/raillabel/test_element_data_pointer.py +++ b/tests/test_raillabel/format/raillabel/test_element_data_pointer.py @@ -9,7 +9,7 @@ sys.path.insert(1, str(Path(__file__).parent.parent.parent.parent.parent)) -from raillabel._util._attribute_type import AttributeType +from raillabel.format.raillabel._attribute_type import AttributeType from raillabel.format import ElementDataPointer # == Fixtures ========================= diff --git a/tests/test_raillabel/format/raillabel/test_object.py b/tests/test_raillabel/format/raillabel/test_object.py index 647289f..bfccad5 100644 --- a/tests/test_raillabel/format/raillabel/test_object.py +++ b/tests/test_raillabel/format/raillabel/test_object.py @@ -12,7 +12,7 @@ sys.path.insert(1, str(Path(__file__).parent.parent.parent.parent.parent)) -from raillabel._util._attribute_type import AttributeType +from raillabel.format.raillabel._attribute_type import AttributeType from raillabel.format import ( Bbox, Cuboid,