Black code formatter

pedrolarben · Jun 13, 2022 · 6ee55d8 · 6ee55d8
1 parent 09c6cff
commit 6ee55d8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -22,7 +22,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install black==21.7b.0 pytest
+        python -m pip install black==22.3.0 pytest
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
     - name: Lint with Black
       run: |

diff --git a/ADLStream/models/transformer.py b/ADLStream/models/transformer.py
@@ -497,7 +497,7 @@ def __init__(self, d_model, warmup_steps=4000):
 
     def __call__(self, step):
         arg1 = tf.math.rsqrt(step)
-        arg2 = step * (self.warmup_steps ** -1.5)
+        arg2 = step * (self.warmup_steps**-1.5)
 
         return tf.math.rsqrt(self.d_model) * tf.math.minimum(arg1, arg2)