add tag

aeon-toolkit · Jun 14, 2024 · 8531816 · 8531816
1 parent 93ba268
commit 8531816
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/aeon/networks/_ae_bgru.py b/aeon/networks/_ae_bgru.py
@@ -22,6 +22,8 @@ class AEBiGRUNetwork(BaseDeepNetwork):
             Flag to choose whether the latent space is an MTS or Euclidean space.
     """
 
+    _tags = {"auto-encoder": True}
+
     def __init__(
         self,
         latent_space_dim=128,