Update test_ae_bgru.py

aeon-toolkit · Jun 4, 2024 · 465b97c · 465b97c
1 parent 93ba268
commit 465b97c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aeon/networks/tests/test_ae_bgru.py b/aeon/networks/tests/test_ae_bgru.py
@@ -49,6 +49,6 @@ def test_aebigrunetwork_activations(activation):
         activation=activation,
         n_units=[50, 50],
     )
-    encoder, decoder = aebigru.build_network((1000, 5))
+    encoder, decoder = aebigru.build_network((150, 5))
     assert encoder is not None
     assert decoder is not None