From 5a7da2b40b108c523b8a16d01ad222443fba9a1b Mon Sep 17 00:00:00 2001
From: Jensun Ravichandran <jjensun@gmail.com>
Date: Tue, 1 Jun 2021 19:03:57 +0200
Subject: [PATCH] [QA] Fix for "no-value-for-parameter" (E1120)

---
 prototorch/modules/losses.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/prototorch/modules/losses.py b/prototorch/modules/losses.py
index 47fe0fb..706d123 100644
--- a/prototorch/modules/losses.py
+++ b/prototorch/modules/losses.py
@@ -20,8 +20,8 @@ class GLVQLoss(torch.nn.Module):
 
 
 class NeuralGasEnergy(torch.nn.Module):
-    def __init__(self, lm):
-        super().__init__()
+    def __init__(self, lm, **kwargs):
+        super().__init__(**kwargs)
         self.lm = lm
 
     def forward(self, d):
@@ -40,8 +40,8 @@ class NeuralGasEnergy(torch.nn.Module):
 
 
 class GrowingNeuralGasEnergy(NeuralGasEnergy):
-    def __init__(self, topology_layer):
-        super().__init__()
+    def __init__(self, topology_layer, **kwargs):
+        super().__init__(**kwargs)
         self.topology_layer = topology_layer
 
     @staticmethod