refactor: move AttributeType

DSD-DBS · Oct 29, 2024 · 3b31c2c · 3b31c2c
1 parent eca43db
commit 3b31c2c
Show file tree

Hide file tree

Showing 7 changed files with 5 additions and 7 deletions.
diff --git 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
diff --git a/raillabel/_util/_attribute_type.py → ...label/format/raillabel/_attribute_type.py b/raillabel/_util/_attribute_type.py → ...label/format/raillabel/_attribute_type.py
diff --git 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
@@ -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
@@ -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
@@ -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,