Refactor: DICOM tags module

jjjermiah · Mar 29, 2024 · 154daad · 154daad
1 parent 0b513e8
commit 154daad
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 2 deletions.
diff --git a/src/nbiatoolkit/dicomtags/__init__.py b/src/nbiatoolkit/dicomtags/__init__.py
@@ -0,0 +1,24 @@
+from .tags import (
+    convert_element_to_int,
+    convert_int_to_element,
+    LOOKUP_TAG,
+    element_VR_lookup,
+    getSeriesModality,
+)
+
+from .tags import (
+    subsetSeriesTags,
+    getReferencedFrameOfReferenceSequence,
+    getReferencedSeriesUIDS,
+)
+
+__all__ = [
+    "convert_element_to_int",
+    "convert_int_to_element",
+    "LOOKUP_TAG",
+    "element_VR_lookup",
+    "getSeriesModality",
+    "subsetSeriesTags",
+    "getReferencedFrameOfReferenceSequence",
+    "getReferencedSeriesUIDS",
+]
diff --git a/src/nbiatoolkit/dicomtags/tags.py b/src/nbiatoolkit/dicomtags/tags.py
@@ -1,8 +1,7 @@
 from pydicom.datadict import dictionary_VR
 from pydicom.datadict import tag_for_keyword
-from pydicom._dicom_dict import DicomDictionary
 import pandas as pd
-from typing import Any, Union, List
+from typing import List
 
 
 def convert_element_to_int(element_str: str) -> int: