ec4516bd0e3df84d3243f6bd0b3395a84fb1c19c,deepchem/data/tests/test_json_loader.py,TestJsonLoader,test_json_loader,#TestJsonLoader#,24

Before Change

    featurizer = SineCoulombMatrix(max_atoms=5)
    loader = JsonLoader(
        tasks=["e_form"],
        json_fields={"structure": dict},
        id_field="formula",
        featurizer=featurizer)
    dataset = loader.create_dataset(input_file, shard_size=1)

After Change

    dataset = loader.create_dataset(input_file, shard_size=None)
    assert dataset.X.shape == (5, 1, 5)

    dataset = loader.create_dataset([input_file, input_file], shard_size=5)
    assert dataset.X.shape == (10, 1, 5)

Instances