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

👌 fix codes by reviews

parent fd602647
Loading
Loading
Loading
Loading
+13 −1
Original line number Diff line number Diff line
@@ -51,11 +51,23 @@ featurizers = {
    'weave': dc.feat.WeaveFeaturizer(),
}

# These featurizers depend on RDKit, so we need RDKit when globally instantiating.
# some featurizers require soft dependencies to instantiate
try:
  featurizers['ecfp'] = dc.feat.CircularFingerprint(size=1024)
except ImportError:
  pass

try:
  featurizers['raw'] = dc.feat.RawFeaturizer()
except ImportError:
  pass

try:
  featurizers['smiles2img'] = dc.feat.SmilesToImage(img_size=80, img_spec='std')
except ImportError:
  pass

try:
  featurizers['onehot'] = dc.feat.OneHotFeaturizer()
except ImportError:
  pass