Fix quality on main (ruff release)

clatonhendricks · Feb 12, 2023 · c836f77 · c836f77
1 parent 75a208e
commit c836f77
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/electra/modeling_flax_electra.py b/src/transformers/models/electra/modeling_flax_electra.py
@@ -1213,7 +1213,7 @@ def setup(self):
             self.summary = nn.Dense(num_classes, dtype=self.dtype)
 
         activation_string = getattr(self.config, "summary_activation", None)
-        self.activation = ACT2FN[activation_string] if activation_string else lambda x: x
+        self.activation = ACT2FN[activation_string] if activation_string else lambda x: x  # noqa F407
 
         self.first_dropout = identity
         if hasattr(self.config, "summary_first_dropout") and self.config.summary_first_dropout > 0: