Commit f43ff4c1 authored by nd-02110114's avatar nd-02110114
Browse files

💚 fix ci error

parent 796c846c

deepchem/data/tests/test_json_loader.py

+4 −4
Original line number Diff line number Diff line
@@ -23,11 +23,11 @@ def test_json_loader(): 


  a = [4625.32086965, 6585.20209678, 61.00680193, 48.72230922, 48.72230922]



  assert dataset.X.shape == (5, 1, 5)

  assert np.allclose(dataset.X[0][0], a, atol=.5)

  assert dataset.X.shape == (5, 5)

  assert np.allclose(dataset.X[0], a, atol=.5)



  dataset = loader.create_dataset(input_file, shard_size=None)

  assert dataset.X.shape == (5, 1, 5)

  assert dataset.X.shape == (5, 5)



  dataset = loader.create_dataset([input_file, input_file], shard_size=5)

  assert dataset.X.shape == (10, 1, 5)
 
  assert dataset.X.shape == (10, 5)

deepchem/feat/tests/test_materials_featurizers.py

+1 −1
Original line number Diff line number Diff line
@@ -68,7 +68,7 @@ class TestMaterialFeaturizers(unittest.TestCase): 


    assert len(features) == 1

    assert features.shape == (1, 3)

    assert np.isclose(features[0], 1244, atol=.5)

    assert np.isclose(features[0][0], 1244, atol=.5)



  def test_cgcnn_featurizer(self):

    """

CRA Git | Maintained and supported by SUSTech CRA and CCSE