Commit 1398ae7d authored by Atreya Majumdar's avatar Atreya Majumdar
Browse files

Changed test decorator

parent fbc8b4c1

deepchem/models/tests/test_layers.py

+1 −1
Original line number Diff line number Diff line
@@ -607,7 +607,7 @@ def test_DAG_gather(): 
  outputs = layer([atom_features, membership])





@pytest.mark.pytorch

@pytest.mark.torch

def test_scale_norm():

  """Test invoking ScaleNorm."""

  input_ar = torch.tensor([[1., 99., 10000.], [0.003, 999.37, 23.]])

CRA Git | Maintained and supported by SUSTech CRA and CCSE