Turn off pandas copy-on-write

closes has2k1/mizani#38
has2k1 · Mar 14, 2024 · 77d9a1f · 77d9a1f
1 parent 1dc71e0
commit 77d9a1f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/doc/changelog.qmd b/doc/changelog.qmd
@@ -2,6 +2,15 @@
 title: Changelog
 ---
 
+## v0.13.2
+(not-yet-released)
+
+### Enhancements
+
+- Drawing the plot ensures that pandas Copy-on-Write is turned off. Plotnine
+  is not yet compatible with Copy-on-Write so this prevents errors for users
+  who have turned on the option.
+
 ## v0.13.1
 (2024-03-06)
 

diff --git a/plotnine/_utils/context.py b/plotnine/_utils/context.py
@@ -39,7 +39,10 @@ def __enter__(self) -> Self:
         # from slices we do not want complaints. We always uses the
         # new frames knowing that they are separate from the original.
         self.pd_option_context = pd.option_context(
-            "mode.chained_assignment", None
+            "mode.chained_assignment",
+            None,
+            "mode.copy_on_write",
+            False,
         )
         self.rc_context.__enter__()
         self.pd_option_context.__enter__()