Unverified Commit ded1a5d2 authored by Suzukazole's avatar Suzukazole
Browse files

fix doctest

parent fa3537c1
Loading
Loading
Loading
Loading
+4 −7
Original line number Diff line number Diff line
@@ -431,17 +431,14 @@ class DummyFeaturizer(Featurizer):
  Examples
  --------
  >>> import deepchem as dc
  >>> smi_map = [["N#C[S-].O=C(CBr)c1ccc(C(F)(F)F)cc1>CCO.[K+]",
  ...             "N#CSCC(=O)c1ccc(C(F)(F)F)cc1"],
  ...           ["C1COCCN1.FCC(Br)c1cccc(Br)n1>CCN(C(C)C)C(C)C.CN(C)C=O.O",
  ...             "FCC(c1cccc(Br)n1)N1CCOCC1"]]
  >>> smi_map = [["N#C[S-].O=C(CBr)c1ccc(C(F)(F)F)cc1>CCO.[K+]", "N#CSCC(=O)c1ccc(C(F)(F)F)cc1"], ["C1COCCN1.FCC(Br)c1cccc(Br)n1>CCN(C(C)C)C(C)C.CN(C)C=O.O", "FCC(c1cccc(Br)n1)N1CCOCC1"]]
  >>> Featurizer = dc.feat.DummyFeaturizer()
  >>> smi_feat = Featurizer.featurize(smi_map)
  >>> smi_feat
  array([['N#C[S-].O=C(CBr)c1ccc(C(F)(F)F)cc1>CCO.[K+]',
  ...     'N#CSCC(=O)c1ccc(C(F)(F)F)cc1'],
  ...    ['C1COCCN1.FCC(Br)c1cccc(Br)n1>CCN(C(C)C)C(C)C.CN(C)C=O.O',
  ...     'FCC(c1cccc(Br)n1)N1CCOCC1']], dtype='<U55')
          'N#CSCC(=O)c1ccc(C(F)(F)F)cc1'],
         ['C1COCCN1.FCC(Br)c1cccc(Br)n1>CCN(C(C)C)C(C)C.CN(C)C=O.O',
          'FCC(c1cccc(Br)n1)N1CCOCC1']], dtype='<U55')
  """

  def featurize(self, datapoints: Iterable[Any],