Modified tags_init and enum_tags file

aeon-toolkit · Dec 17, 2024 · 0128569 · 0128569
1 parent fc67481
commit 0128569
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/aeon/utils/tags/__init__.py b/aeon/utils/tags/__init__.py
@@ -1,6 +1,7 @@
 """Estimator tags and tag utilities."""
 
 __all__ = [
+    "AlgorithmType",
     "ESTIMATOR_TAGS",
     "check_valid_tags",
     "all_tags_for_estimator",
@@ -9,3 +10,4 @@
 from aeon.utils.tags._discovery import all_tags_for_estimator
 from aeon.utils.tags._tags import ESTIMATOR_TAGS
 from aeon.utils.tags._validate import check_valid_tags
+from aeon.utils.tags.enum_tags import AlgorithmType
diff --git a/aeon/utils/tags/enum_tags.py b/aeon/utils/tags/enum_tags.py
@@ -1,5 +1,7 @@
 """Apply Enumeration in module."""
 
+__all__ = ["AlgorithmType"]
+
 from enum import Enum