From db2b488311a5f74ba8af326398e3cb2d4de2ebb0 Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Wed, 16 Sep 2026 05:50:36 -0700 Subject: [PATCH] suppress pyrefly errors related to JAX scalar types PiperOrigin-RevId: 982463549 --- edward2/jax/nn/random_feature.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/edward2/jax/nn/random_feature.py b/edward2/jax/nn/random_feature.py index c8277d06..64670762 100644 --- a/edward2/jax/nn/random_feature.py +++ b/edward2/jax/nn/random_feature.py @@ -162,7 +162,7 @@ class RandomFourierFeatures(nn.Module): dtype: the dtype of the computation (default: float32). """ features: int - feature_scale: Optional[jnp.float32] = 1. + feature_scale: Optional[jnp.float32] = 1. # pyrefly: ignore[not-a-type] activation: Callable[[Array], Array] = default_rbf_activation kernel_init: Initializer = default_rbf_kernel_init bias_init: Initializer = default_rbf_bias_init