changed keywords

SURGroup · Aug 1, 2024 · 54cf6a8 · 54cf6a8
1 parent 60f16ab
commit 54cf6a8
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/UQpy/scientific_machine_learning/baseclass/Loss.py b/src/UQpy/scientific_machine_learning/baseclass/Loss.py
@@ -3,9 +3,8 @@
 
 
 class Loss(nn.Module, ABC):
-    def __init__(self, **kwargs):
-        super().__init__(**kwargs)
+    def __init__(self):
+        super().__init__()
 
     @abstractmethod
-    def forward(self):
-        ...
+    def forward(self, *args, **kwargs): ...