expose pygam model penalties via spline_kwargs

theislab · Oct 20, 2023 · b8038a7 · b8038a7
1 parent c3ced63
commit b8038a7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/cellrank/models/_pygam_model.py b/src/cellrank/models/_pygam_model.py
@@ -106,8 +106,7 @@ def __init__(
             0,
             spline_order=spline_order,
             n_splines=n_knots,
-            penalties=["derivative", "l2"],
-            **_filter_kwargs(s, **{**{"lam": 3}, **spline_kwargs}),
+            **_filter_kwargs(s, **{**{"lam": 3, "penalties": ["derivative", "l2"]}, **spline_kwargs}),
         )
         link = GamLinkFunction(link)
         distribution = GamDistribution(distribution)