Loading deepchem/models/tests/test_layers.py +1 −1 Original line number Diff line number Diff line Loading @@ -612,7 +612,7 @@ def test_scale_norm(): """Test invoking ScaleNorm.""" input_ar = torch.tensor([[1., 99., 10000.], [0.003, 999.37, 23.]]) layer = torch_layers.ScaleNorm(0.35) result1 = layer.forward(input_ar) result1 = layer.forward(input_ar).detach().numpy() output_ar = np.array([[5.9157897e-05, 5.8566318e-03, 5.9157896e-01], [1.7754727e-06, 5.9145141e-01, 1.3611957e-02]]) assert np.allclose(result1, output_ar) Loading
deepchem/models/tests/test_layers.py +1 −1 Original line number Diff line number Diff line Loading @@ -612,7 +612,7 @@ def test_scale_norm(): """Test invoking ScaleNorm.""" input_ar = torch.tensor([[1., 99., 10000.], [0.003, 999.37, 23.]]) layer = torch_layers.ScaleNorm(0.35) result1 = layer.forward(input_ar) result1 = layer.forward(input_ar).detach().numpy() output_ar = np.array([[5.9157897e-05, 5.8566318e-03, 5.9157896e-01], [1.7754727e-06, 5.9145141e-01, 1.3611957e-02]]) assert np.allclose(result1, output_ar)
