diff --git a/aeon/networks/_ae_bgru.py b/aeon/networks/_ae_bgru.py index c1bdf781d8..0fa91aefd1 100644 --- 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,