Update sqrt and ref kwargs

alvgaona · alvgaona · commit f5e1c45ad5b0 · 2022-10-11T14:15:09.000-03:00
diff --git a/test/test_signal.py b/test/test_signal.py
@@ -32,7 +32,8 @@ def _test_window(self, device, dtype, op: OpInfo, **kwargs):
             window_name = re.search(self.supported_windows, op.name).group(0)
 
             ref_kwargs = {
-                k: sample_input.kwargs[k] for k in sample_input.kwargs if k not in ('device', 'dtype', 'requires_grad', 'periodic')
+                k: sample_input.kwargs[k] for k in sample_input.kwargs
+                if k not in ('device', 'dtype', 'requires_grad', 'periodic')
             }
 
             expected = torch.from_numpy(
diff --git a/torch/signal/windows/windows.py b/torch/signal/windows/windows.py
@@ -1,5 +1,4 @@
 import torch
-import numpy as np
 
 from torch import Tensor
 from torch._torch_docs import factory_common_args
@@ -299,7 +298,7 @@ def gaussian(window_length: int,
 
     start = -(window_length if periodic else window_length - 1) / 2.0
 
-    constant = 1 / (std * np.sqrt(2))
+    constant = 1 / (std * torch.sqrt(torch.tensor(2)).item())
 
     """
     Note that non-integer step is subject to floating point rounding errors when comparing against end;

Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,8 @@ def _test_window(self, device, dtype, op: OpInfo, **kwargs):`
`32`	`32`	`window_name = re.search(self.supported_windows, op.name).group(0)`
`33`	`33`
`34`	`34`	`ref_kwargs = {`
`35`		`- k: sample_input.kwargs[k] for k in sample_input.kwargs if k not in ('device', 'dtype', 'requires_grad', 'periodic')`
	`35`	`+ k: sample_input.kwargs[k] for k in sample_input.kwargs`
	`36`	`+ if k not in ('device', 'dtype', 'requires_grad', 'periodic')`
`36`	`37`	`}`
`37`	`38`
`38`	`39`	`expected = torch.from_numpy(`