images regen (#98)

* images regen * lint
vincentarelbundock · May 1, 2024 · 85394c4 · 85394c4
1 parent a8f8a41
commit 85394c4
Show file tree

Hide file tree

Showing 23 changed files with 5 additions and 3 deletions.
diff --git a/marginaleffects/comparisons.py b/marginaleffects/comparisons.py
@@ -276,7 +276,9 @@ def applyfun(x, by, wts=None):
 
         # maintain_order is extremely important
         by = [x for x in by if x in tmp.columns]
-        tmp = tmp.group_by(*by, maintain_order=True).map_groups(function=lambda x: applyfun(x, by=by, wts=wts))
+        tmp = tmp.group_by(*by, maintain_order=True).map_groups(
+            function=lambda x: applyfun(x, by=by, wts=wts)
+        )
 
         tmp = get_hypothesis(tmp, hypothesis=hypothesis)
 

diff --git a/tests/images/jss/jss_01.png b/tests/images/jss/jss_01.png
diff --git a/tests/images/plot_comparisons/continous_01.png b/tests/images/plot_comparisons/continous_01.png
diff --git a/tests/images/plot_comparisons/continous_02.png b/tests/images/plot_comparisons/continous_02.png
diff --git a/tests/images/plot_comparisons/continuous_01.png b/tests/images/plot_comparisons/continuous_01.png
diff --git a/tests/images/plot_comparisons/continuous_02.png b/tests/images/plot_comparisons/continuous_02.png
diff --git a/tests/images/plot_comparisons/continuous_03.png b/tests/images/plot_comparisons/continuous_03.png
diff --git a/tests/images/plot_comparisons/discrete_01.png b/tests/images/plot_comparisons/discrete_01.png
diff --git a/tests/images/plot_comparisons/discrete_02.png b/tests/images/plot_comparisons/discrete_02.png
diff --git a/tests/images/plot_comparisons/discrete_03.png b/tests/images/plot_comparisons/discrete_03.png
diff --git a/tests/images/plot_predictions/by_01.png b/tests/images/plot_predictions/by_01.png
diff --git a/tests/images/plot_predictions/by_02.png b/tests/images/plot_predictions/by_02.png
diff --git a/tests/images/plot_predictions/condition_01.png b/tests/images/plot_predictions/condition_01.png
diff --git a/tests/images/plot_predictions/condition_02.png b/tests/images/plot_predictions/condition_02.png
diff --git a/tests/images/plot_predictions/issue_57_01.png b/tests/images/plot_predictions/issue_57_01.png
diff --git a/tests/images/plot_predictions/issue_57_02.png b/tests/images/plot_predictions/issue_57_02.png
diff --git a/tests/images/plot_predictions/issue_57_03.png b/tests/images/plot_predictions/issue_57_03.png
diff --git a/tests/images/plot_predictions/issue_57_04.png b/tests/images/plot_predictions/issue_57_04.png
diff --git a/tests/images/plot_slopes/condition_01.png b/tests/images/plot_slopes/condition_01.png
diff --git a/tests/images/plot_slopes/condition_02.png b/tests/images/plot_slopes/condition_02.png
diff --git a/tests/images/plot_slopes/condition_03.png b/tests/images/plot_slopes/condition_03.png
diff --git a/tests/images/plot_slopes/condition_04.png b/tests/images/plot_slopes/condition_04.png
diff --git a/tests/test_plot_comparisons.py b/tests/test_plot_comparisons.py
@@ -27,12 +27,12 @@ def test_continuous():
         variables="bill_length_mm",
         by="island",
     )
-    assert assert_image(fig, "continous_01", "plot_comparisons") is None
+    assert assert_image(fig, "continuous_01", "plot_comparisons") is None
 
     fig = plot_comparisons(
         mod, variables="bill_length_mm", condition=["flipper_length_mm", "species"]
     )
-    assert assert_image(fig, "continous_02", "plot_comparisons") is None
+    assert assert_image(fig, "continuous_02", "plot_comparisons") is None
 
     fig = plot_comparisons(mod, variables="bill_length_mm", condition="species")
     assert assert_image(fig, "continuous_03", "plot_comparisons") is None