update base

aeon-toolkit · Jun 21, 2024 · 1cc049c · 1cc049c
1 parent f0895e1
commit 1cc049c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/aeon/networks/_ae_bgru.py b/aeon/networks/_ae_bgru.py
@@ -1,9 +1,9 @@
 """Implement Auto-Encoder based on Bidirectional GRUs."""
 
-from aeon.networks.base import BaseDeepNetwork
+from aeon.networks.base import BaseDeepLearningNetwork
 
 
-class AEBiGRUNetwork(BaseDeepNetwork):
+class AEBiGRUNetwork(BaseDeepLearningNetwork):
     """
     A class to implement an Auto-Encoder based on Bidirectional GRUs.
 
@@ -22,7 +22,7 @@ class AEBiGRUNetwork(BaseDeepNetwork):
             Flag to choose whether the latent space is an MTS or Euclidean space.
     """
 
-    _tags = {"auto-encoder": True}
+    _config["auto-encoder"]=True
 
     def __init__(
         self,