[QA] Fix for "no-value-for-parameter" (E1120)
This commit is contained in:
parent
b6d38f442b
commit
5a7da2b40b
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user