julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[QA] Fix for "dangerous-default-value" (W0102)

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-06-01 19:24:00 +02:00
parent 2c453265fe
commit e8e803e8ef
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
prototorch/components/initializers.py
View File

@ -155,7 +155,7 @@ class StratifiedSelectionInitializer(ClassAwareInitializer):
mask = torch.bernoulli(n1) - torch.bernoulli(n2) mask = torch.bernoulli(n1) - torch.bernoulli(n2)
return x + (self.noise * mask) return x + (self.noise * mask)
def generate(self, length, dist=[]): def generate(self, length, dist):
samples = self._get_samples_from_initializer(length, dist) samples = self._get_samples_from_initializer(length, dist)
if self.noise is not None: if self.noise is not None:
samples = self.add_noise_v1(samples) samples = self.add_noise_v1(samples)