Clean-up code (#4790)

acaa2e62 · Sylvain Gugger · GitHub · fa661ce7 · acaa2e62
Unverified Commit acaa2e62 authored Jun 05, 2020 by Sylvain Gugger Committed by GitHub Jun 05, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

src/transformers/configuration_xlnet.py src/transformers/configuration_xlnet.py +3 -3

No files found.
--- a/src/transformers/configuration_xlnet.py
+++ b/src/transformers/configuration_xlnet.py
@@ -164,9 +164,9 @@ class XLNetConfig(PretrainedConfig):
        self.n_head = n_head
        assert d_model % n_head == 0
        if "d_head" in kwargs:
-            assert kwargs["d_head"] == d_model // n_head, (
+            assert (
-                f"`d_head` ({kwargs['d_head']}) should be equal to `d_model // n_head` ({d_model // n_head})"
+                kwargs["d_head"] == d_model // n_head
-            )
+            ), f"`d_head` ({kwargs['d_head']}) should be equal to `d_model // n_head` ({d_model // n_head})"
        self.d_head = d_model // n_head
        self.ff_activation = ff_activation
        self.d_inner = d_inner