From 20f0a2ce75beba6b679d5027f3af721586981102 Mon Sep 17 00:00:00 2001 From: larskue Date: Mon, 14 Oct 2024 10:26:10 +0200 Subject: [PATCH] add type hints --- bayesflow/networks/transformers/isab.py | 2 +- bayesflow/networks/transformers/mab.py | 2 +- bayesflow/networks/transformers/pma.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/bayesflow/networks/transformers/isab.py b/bayesflow/networks/transformers/isab.py index 529154b6..5bf04928 100644 --- 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 index 155e925b..a96587df 100644 --- 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 index 9c95ee15..dbd7648b 100644 --- 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, ):