Commit 14e7e8d3 authored by peastman's avatar peastman
Browse files

Fixed mypy errors

parent 2b665347
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -2,6 +2,7 @@ import numpy as np

from deepchem.utils.typing import PymatgenComposition
from deepchem.feat import MaterialCompositionFeaturizer
from typing import Any


class ElementPropertyFingerprint(MaterialCompositionFeaturizer):
@@ -51,7 +52,7 @@ class ElementPropertyFingerprint(MaterialCompositionFeaturizer):
      Source for element property data.
    """
    self.data_source = data_source
    self.ep_featurizer = None
    self.ep_featurizer: Any = None

  def _featurize(self, composition: PymatgenComposition) -> np.ndarray:
    """
+2 −1
Original line number Diff line number Diff line
@@ -3,6 +3,7 @@ import numpy as np
from deepchem.utils.typing import PymatgenStructure
from deepchem.feat import MaterialStructureFeaturizer
from deepchem.utils.data_utils import pad_array
from typing import Any


class SineCoulombMatrix(MaterialStructureFeaturizer):
@@ -56,7 +57,7 @@ class SineCoulombMatrix(MaterialStructureFeaturizer):
    """
    self.max_atoms = max_atoms
    self.flatten = flatten
    self.scm = None
    self.scm: Any = None

  def _featurize(self, struct: PymatgenStructure) -> np.ndarray:
    """