Merge pull request #694 from molpopgen/more_type_hint_fixes

More type hint fixes in DiploidPopulation.
molpopgen · Mar 6, 2021 · 7e0c62b · 7e0c62b
2 parents 76c05d0 + 4d262cb
commit 7e0c62b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fwdpy11/_types/diploid_population.py b/fwdpy11/_types/diploid_population.py
@@ -67,7 +67,7 @@ def create_from_tskit(cls, ts: tskit.TreeSequence) -> "fwdpy11.DiploidPopulation
         return cls(0, 0.0, ll_pop=ll)
 
     @classmethod
-    def load_from_file(cls, filename: IO) -> "fwdpy11.DiploidPopulation":
+    def load_from_file(cls, filename: str) -> "fwdpy11.DiploidPopulation":
         """
         Load population from the output of
         :func:`fwdpy11.DiploidPopulation.dump_to_file`.
@@ -208,7 +208,7 @@ def dump_to_file(self, filename: str):
         """
         self._dump_to_file(filename)
 
-    def pickle_to_file(self, filename: str):
+    def pickle_to_file(self, filename: IO):
         """
         Pickle the population to an open file.