Update _mlp.py

aeon-toolkit · Nov 11, 2024 · ceea333 · ceea333
1 parent 3aa92a0
commit ceea333
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/aeon/networks/_mlp.py b/aeon/networks/_mlp.py
@@ -3,7 +3,7 @@
 __maintainer__ = ["hadifawaz1999"]
 
 
-from typing import List, Union
+from typing import Union
 
 from aeon.networks.base import BaseDeepLearningNetwork
 
@@ -17,11 +17,11 @@ class MLPNetwork(BaseDeepLearningNetwork):
     ----------
     n_layers : int, optional (default=3)
         The number of dense layers in the MLP.
-    n_units : Union[int, List[int]], optional (default=200)
+    n_units : Union[int, list[int]], optional (default=200)
         Number of units in each dense layer.
-    activation : Union[str, List[str]], optional (default='relu')
+    activation : Union[str, list[str]], optional (default='relu')
         Activation function(s) for each dense layer.
-    dropout_rate : Union[int, float, List[Union[int, float]]], optional (default=None)
+    dropout_rate : Union[int, float, list[Union[int, float]]], optional (default=None)
         Dropout rate(s) for each dense layer. If None, a default rate of 0.2 is used.
 
     Notes