Silence spurious pytype error due to attr inheritance

google-deepmind · Nov 6, 2023 · d5c8966 · d5c8966
1 parent 21a3953
commit d5c8966
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ferminet/psiformer.py b/ferminet/psiformer.py
@@ -401,7 +401,7 @@ def make_fermi_net(
       heads_dim=heads_dim,
       mlp_hidden_dims=mlp_hidden_dims,
       use_layer_norm=use_layer_norm,
-  )
+  )  # pytype: disable=wrong-keyword-args
 
   psiformer_layers = make_psiformer_layers(nspins, charges.shape[0], options)