fmt and fix sim time

molpopgen · Nov 3, 2023 · 2400cdd · 2400cdd
1 parent 3da9d81
commit 2400cdd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/examples/gss_divergent_optima/gss_divergent_optima.py b/examples/gss_divergent_optima/gss_divergent_optima.py
@@ -23,7 +23,8 @@
 """
 
 demography = fwdpy11.ForwardDemesGraph.from_demes(
-    yaml, burnin=10*N, burnin_is_exact=True)
+    yaml, burnin=10 * N + 200, burnin_is_exact=True
+)
 
 moving_optimum_deme_0 = fwdpy11.GSSmo(
     [
@@ -70,16 +71,15 @@
             # Mapping of trait (genetic) value to fitness
             gvalue_to_fitness=moving_optimum_deme_0,
         ),
-        fwdpy11.Additive(ndemes=2, scaling=2,
-                         gvalue_to_fitness=moving_optimum_deme_1),
+        fwdpy11.Additive(ndemes=2, scaling=2, gvalue_to_fitness=moving_optimum_deme_1),
     ],
 }
 
 params = fwdpy11.ModelParams(**pdict)
 pop = fwdpy11.DiploidPopulation([N, N], 1.0)
 rng = fwdpy11.GSLrng(1010)
 fwdpy11.evolvets(rng, pop, params, 100)
-assert pop.generation == 10*N + 200
+assert pop.generation == 10 * N + 200
 md = np.array(pop.diploid_metadata, copy=False)
 df = pd.DataFrame.from_records(md[["deme", "g", "w"]])
 g = df.groupby(["deme"])