[QA] Fix for "no-value-for-parameter" (E1120)

This commit is contained in:
Jensun Ravichandran 2021-06-01 19:03:57 +02:00
parent b6d38f442b
commit 5a7da2b40b

View File

@ -20,8 +20,8 @@ class GLVQLoss(torch.nn.Module):
class NeuralGasEnergy(torch.nn.Module): class NeuralGasEnergy(torch.nn.Module):
def __init__(self, lm): def __init__(self, lm, **kwargs):
super().__init__() super().__init__(**kwargs)
self.lm = lm self.lm = lm
def forward(self, d): def forward(self, d):
@ -40,8 +40,8 @@ class NeuralGasEnergy(torch.nn.Module):
class GrowingNeuralGasEnergy(NeuralGasEnergy): class GrowingNeuralGasEnergy(NeuralGasEnergy):
def __init__(self, topology_layer): def __init__(self, topology_layer, **kwargs):
super().__init__() super().__init__(**kwargs)
self.topology_layer = topology_layer self.topology_layer = topology_layer
@staticmethod @staticmethod