From bdda28c77798d3c08913fd9c9059710f288e0e41 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 5 Jun 2022 21:20:16 +0200
Subject: Fix kwargs

---
 text_recognizer/networks/conformer/scale.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'text_recognizer/networks')

diff --git a/text_recognizer/networks/conformer/scale.py b/text_recognizer/networks/conformer/scale.py
index da736a3..d012b81 100644
--- a/text_recognizer/networks/conformer/scale.py
+++ b/text_recognizer/networks/conformer/scale.py
@@ -9,5 +9,5 @@ class Scale(nn.Module):
         self.scale = scale
         self.fn = fn
 
-    def forward(self, x: Tensor, kwargs: Dict) -> Tensor:
+    def forward(self, x: Tensor, **kwargs) -> Tensor:
         return self.fn(x, **kwargs) * self.scale
-- 
cgit v1.2.3-70-g09d2