Commit fbc8b4c1 authored by Atreya Majumdar's avatar Atreya Majumdar
Browse files

Changed Docstring

parent f925df0f

deepchem/models/tests/test_layers.py

+1 −1
Original line number Diff line number Diff line
@@ -609,7 +609,7 @@ def test_DAG_gather(): 


@pytest.mark.pytorch

def test_scale_norm():

  """Test invoking LayerNorm."""

  """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)

CRA Git | Maintained and supported by SUSTech CRA and CCSE