Merge pull request #19 from ecrl/20250122_typehints

Update type hints
ecrl · Jan 23, 2025 · c1d0091 · c1d0091
2 parents 3030031 + ab0fe77
commit c1d0091
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/graphchem/preprocessing/features.py b/graphchem/preprocessing/features.py
@@ -7,7 +7,7 @@
 
 
 def get_ring_size(
-        obj: Union[rdkit.Chem.Atom, rdkit.Chem.Bond],
+        obj: Union["rdkit.Chem.Atom", "rdkit.Chem.Bond"],
         max_size: Optional[int] = 12
      ) -> int:
     """
@@ -35,7 +35,7 @@ def get_ring_size(
     return max_size
 
 
-def atom_to_str(atom: rdkit.Chem.Atom) -> str:
+def atom_to_str(atom: "rdkit.Chem.Atom") -> str:
     """
     Convert an RDKit Atom object to a string representation.
 
@@ -82,7 +82,7 @@ def atom_to_str(atom: rdkit.Chem.Atom) -> str:
     ))
 
 
-def bond_to_str(bond: rdkit.Chem.Bond) -> str:
+def bond_to_str(bond: "rdkit.Chem.Bond") -> str:
     """
     Convert an RDKit Bond object to a string representation.
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -7,7 +7,7 @@ include-package-data = true
 
 [project]
 name = "graphchem"
-version = "2.3.1"
+version = "2.3.2"
 authors = [
     { name="Travis Kessler", email="[email protected]" },
 ]