Commit 7f5297b6 authored by Nathan Frey's avatar Nathan Frey
Browse files

Remove type hints

parent ab558d41

deepchem/molnet/load_function/material_datasets/load_bandgap.py

+1 −1
Original line number Diff line number Diff line
@@ -155,7 +155,7 @@ def load_bandgap( 
      return my_tasks, all_dataset, transformers



  # First type of supported featurizers

  supported_featurizers = ['ElementPropertyFingerprint']  # type: List[str]

  supported_featurizers: List[str] = ['ElementPropertyFingerprint']



  # Load .tar.gz file

  if featurizer.__class__.__name__ in supported_featurizers:

deepchem/molnet/load_function/material_datasets/load_perovskite.py

+3 −2
Original line number Diff line number Diff line
@@ -153,8 +153,9 @@ def load_perovskite( 
      return my_tasks, all_dataset, transformers



  # First type of supported featurizers

  supported_featurizers = ['StructureGraphFeaturizer',

                           'SineCoulombMatrix']  # type: List[str]

  supported_featurizers: List[str] = [

      'StructureGraphFeaturizer', 'SineCoulombMatrix'

  ]



  # Load .tar.gz file

  if featurizer.__class__.__name__ in supported_featurizers:

CRA Git | Maintained and supported by SUSTech CRA and CCSE