Commit 4370a148 authored by nd-02110114's avatar nd-02110114
Browse files

🚨 fix yapf

parent 86b384d8
Loading
Loading
Loading
Loading
+0 −1
Original line number Diff line number Diff line
@@ -81,4 +81,3 @@ class ElementPropertyFingerprint(MaterialCompositionFeaturizer):
      feats = []

    return np.nan_to_num(np.array(feats))
+2 −4
Original line number Diff line number Diff line
@@ -22,7 +22,7 @@ PEROVSKITE_URL = 'http://deepchem.io.s3-website-us-west-1.amazonaws.com/datasets
DEFAULT_FEATURIZERS = get_defaults("feat")

# Names of supported featurizers
featurizers = ['SineCoulombMatrix', 'StructureGraphFeaturizer']
featurizers = ['SineCoulombMatrix', 'CGCNNFeaturizer']
DEFAULT_FEATURIZERS = {k: DEFAULT_FEATURIZERS[k] for k in featurizers}

# dict of accepted transformers
@@ -153,9 +153,7 @@ def load_perovskite(
      return my_tasks, all_dataset, transformers

  # First type of supported featurizers
  supported_featurizers: List[str] = [
      'StructureGraphFeaturizer', 'SineCoulombMatrix'
  ]
  supported_featurizers: List[str] = ['CGCNNFeaturizer', 'SineCoulombMatrix']

  # Load .tar.gz file
  if featurizer.__class__.__name__ in supported_featurizers: