diff --git a/Scripts/gen_rdkit_stubs/patch/rdkit/Chem/__init__.pyi.diff b/Scripts/gen_rdkit_stubs/patch/rdkit/Chem/__init__.pyi.diff new file mode 100644 index 000000000..d53a79e17 --- /dev/null +++ b/Scripts/gen_rdkit_stubs/patch/rdkit/Chem/__init__.pyi.diff @@ -0,0 +1,10 @@ +--- a/rdkit/Chem/__init__.pyi ++++ b/rdkit/Chem/__init__.pyi +@@ -138,7 +138,7 @@ + ... + def _sizeCalc(self): + ... +-def CanonSmiles(smi, useChiral = 1): ++def CanonSmiles(smi: str, useChiral: int = 1) -> str: + """ + A convenience function for canonicalizing SMILES diff --git a/Scripts/gen_rdkit_stubs/patch/rdkit/Chem/rdchem.pyi.diff b/Scripts/gen_rdkit_stubs/patch/rdkit/Chem/rdchem.pyi.diff new file mode 100644 index 000000000..100d4fc38 --- /dev/null +++ b/Scripts/gen_rdkit_stubs/patch/rdkit/Chem/rdchem.pyi.diff @@ -0,0 +1,20 @@ +--- a/rdkit/Chem/rdchem.pyi ++++ b/rdkit/Chem/rdchem.pyi +@@ -1988,7 +1988,7 @@ + C++ signature : + RDKit::Atom* GetAtomWithIdx(RDKit::ROMol {lvalue},unsigned int) + """ +- def GetAtoms(self): ++ def GetAtoms(self) -> typing.Iterable[Atom]: + """ + returns an iterator over the atoms in the molecule + """ +@@ -2030,7 +2030,7 @@ + C++ signature : + RDKit::Bond* GetBondWithIdx(RDKit::ROMol {lvalue},unsigned int) + """ +- def GetBonds(self): ++ def GetBonds(self) -> typing.Iterable[Bond]: + """ + returns an iterator over the bonds in the molecule + """