add type hints

bayesflow-org · Oct 14, 2024 · 20f0a2c · 20f0a2c
1 parent c46e706
commit 20f0a2c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/bayesflow/networks/transformers/isab.py b/bayesflow/networks/transformers/isab.py
@@ -26,7 +26,7 @@ def __init__(
         dense_units: int = 128,
         dense_activation: str = "gelu",
         kernel_initializer: str = "he_normal",
-        use_bias=True,
+        use_bias: bool = True,
         layer_norm: bool = True,
         **kwargs,
     ):

diff --git a/bayesflow/networks/transformers/mab.py b/bayesflow/networks/transformers/mab.py
@@ -25,7 +25,7 @@ def __init__(
         dense_units: int = 128,
         dense_activation: str = "gelu",
         kernel_initializer: str = "he_normal",
-        use_bias=True,
+        use_bias: bool = True,
         layer_norm: bool = True,
         **kwargs,
     ):

diff --git a/bayesflow/networks/transformers/pma.py b/bayesflow/networks/transformers/pma.py
@@ -31,7 +31,7 @@ def __init__(
         dense_units: int = 128,
         dense_activation: str = "gelu",
         kernel_initializer: str = "he_normal",
-        use_bias=True,
+        use_bias: bool = True,
         layer_norm: bool = True,
         **kwargs,
     ):