Commit 833a02b6 authored by alat-rights's avatar alat-rights
Browse files

numpy -> np

parent 54d9e3c3
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -72,7 +72,7 @@ class OneHotFeaturizer(Featurizer):

  def _featurize(self, datapoint: Any):
    # Featurize str data
    if (isinstance(datapoint, (str, numpy.str_))):
    if (isinstance(datapoint, (str, np.str_))):
      return self._featurize_string(datapoint)
    # Featurize mol data
    else: