add type hints

fwdpy11/_functions/import_demes.py

@@ -2,6 +2,7 @@
     Dict,
     List,
     Optional,
+    Union,
 )
 import numpy as np
 import attr
@@ -30,7 +31,9 @@
 from .._demography import exponential_growth_rate
 
 # TODO: need type hints for dg
-def demography_from_demes(dg, burnin) -> DemographicModelDetails:
+def demography_from_demes(
+    dg: Union[str, demes.Graph], burnin: int
+) -> DemographicModelDetails:
     """
     The deme graph, dg, can be either a string or a resolved deme-graph.
 
@@ -465,7 +468,13 @@ def _process_epoch(
             )
 
 
-def _process_all_epochs(dg, idmap, model_times, burnin_generation, events):
+def _process_all_epochs(
+    dg: demes.Graph,
+    idmap: Dict,
+    model_times: _ModelTimes,
+    burnin_generation: int,
+    events: _Fwdpy11Events,
+):
     """
     Processes all epochs of all demes to set sizes and selfing rates.
     """

fwdpy11/discrete_demography.py

@@ -23,6 +23,7 @@
 import numpy as np
 
 import fwdpy11
+import demes
 
 from .class_decorators import (
     attr_add_asblack,
@@ -526,7 +527,9 @@ def _timed_events(self):
             yield i
 
 
-def from_demes(dg, burnin=10) -> "DemographicModelDetails":
+def from_demes(
+    dg: typing.Union[str, demes.Graph], burnin: int = 10
+) -> "DemographicModelDetails":
     """
     Build a :class:`fwdpy11.DiscreteDemography` object using
     demes.