Revert "Move chained assignment detection to cython for Python 3.14 compat (pandas-dev#62070)"

This reverts commit a505423.

Author: jorisvandenbossche

pandas/_libs/internals.pyi

@@ -94,7 +94,3 @@ class BlockValuesRefs:
     def add_reference(self, blk: Block) -> None: ...
     def add_index_reference(self, index: Index) -> None: ...
     def has_reference(self) -> bool: ...
-
-class SetitemMixin:
-    def __setitem__(self, key, value) -> None: ...
-    def __delitem__(self, key) -> None: ...

pandas/_libs/internals.pyx

@@ -1,9 +1,6 @@
 from collections import defaultdict
-import sys
-import warnings
 
 cimport cython
-from cpython cimport PY_VERSION_HEX
 from cpython.object cimport PyObject
 from cpython.pyport cimport PY_SSIZE_T_MAX
 from cpython.slice cimport PySlice_GetIndicesEx
@@ -23,9 +20,6 @@ from numpy cimport (
 cnp.import_array()
 
 from pandas._libs.algos import ensure_int64
-from pandas.compat import CHAINED_WARNING_DISABLED
-from pandas.errors import ChainedAssignmentError
-from pandas.errors.cow import _chained_assignment_msg
 
 from pandas._libs.util cimport (
     is_array,
@@ -1002,47 +996,3 @@ cdef class BlockValuesRefs:
                 return self._has_reference_maybe_locked()
         ELSE:
             return self._has_reference_maybe_locked()
-
-
-cdef extern from "Python.h":
-    """
-    // python version < 3.14
-    #if PY_VERSION_HEX < 0x030E0000
-    // This function is unused and is declared to avoid a build warning
-    int __Pyx_PyUnstable_Object_IsUniqueReferencedTemporary(PyObject *ref) {
-        return Py_REFCNT(ref) == 1;
-    }
-    #else
-    #define __Pyx_PyUnstable_Object_IsUniqueReferencedTemporary \
-        PyUnstable_Object_IsUniqueReferencedTemporary
-    #endif
-    """
-    int PyUnstable_Object_IsUniqueReferencedTemporary\
-        "__Pyx_PyUnstable_Object_IsUniqueReferencedTemporary"(object o) except -1
-
-
-# Python version compatibility for PyUnstable_Object_IsUniqueReferencedTemporary
-cdef inline bint _is_unique_referenced_temporary(object obj) except -1:
-    if PY_VERSION_HEX >= 0x030E0000:
-        # Python 3.14+ has PyUnstable_Object_IsUniqueReferencedTemporary
-        return PyUnstable_Object_IsUniqueReferencedTemporary(obj)
-    else:
-        # Fallback for older Python versions using sys.getrefcount
-        return sys.getrefcount(obj) <= 1
-
-
-cdef class SetitemMixin:
-    # class used in DataFrame and Series for checking for chained assignment
-
-    def __setitem__(self, key, value) -> None:
-        cdef bint is_unique = 0
-        if not CHAINED_WARNING_DISABLED:
-            is_unique = _is_unique_referenced_temporary(self)
-            if is_unique:
-                warnings.warn(
-                    _chained_assignment_msg, ChainedAssignmentError, stacklevel=1
-                )
-        self._setitem(key, value)
-
-    def __delitem__(self, key) -> None:
-        self._delitem(key)

pandas/_testing/contexts.py

@@ -13,8 +13,8 @@
 import uuid
 
 from pandas.compat import (
-    CHAINED_WARNING_DISABLED,
-    CHAINED_WARNING_DISABLED_INPLACE_METHOD,
+    PYPY,
+    WARNING_CHECK_DISABLED,
 )
 from pandas.errors import ChainedAssignmentError
 
@@ -163,18 +163,10 @@ def with_csv_dialect(name: str, **kwargs) -> Generator[None]:
         csv.unregister_dialect(name)
 
 
-def raises_chained_assignment_error(
-    extra_warnings=(), extra_match=(), inplace_method=False
-):
+def raises_chained_assignment_error(extra_warnings=(), extra_match=()):
     from pandas._testing import assert_produces_warning
 
-    WARNING_DISABLED = (
-        CHAINED_WARNING_DISABLED_INPLACE_METHOD
-        if inplace_method
-        else CHAINED_WARNING_DISABLED
-    )
-
-    if WARNING_DISABLED:
+    if PYPY or WARNING_CHECK_DISABLED:
         if not extra_warnings:
             from contextlib import nullcontext
 

pandas/compat/__init__.py

@@ -16,13 +16,12 @@
 from typing import TYPE_CHECKING
 
 from pandas.compat._constants import (
-    CHAINED_WARNING_DISABLED,
-    CHAINED_WARNING_DISABLED_INPLACE_METHOD,
     IS64,
     ISMUSL,
     PY312,
     PY314,
     PYPY,
+    WARNING_CHECK_DISABLED,
     WASM,
 )
 from pandas.compat.numpy import is_numpy_dev
@@ -153,15 +152,14 @@ def is_ci_environment() -> bool:
 
 
 __all__ = [
-    "CHAINED_WARNING_DISABLED",
-    "CHAINED_WARNING_DISABLED_INPLACE_METHOD",
     "HAS_PYARROW",
     "IS64",
     "ISMUSL",
     "PY312",
     "PY314",
     "PYARROW_MIN_VERSION",
     "PYPY",
+    "WARNING_CHECK_DISABLED",
     "WASM",
     "is_numpy_dev",
     "pa_version_under14p0",

pandas/compat/_constants.py

@@ -19,8 +19,7 @@
 WASM = (sys.platform == "emscripten") or (platform.machine() in ["wasm32", "wasm64"])
 ISMUSL = "musl" in (sysconfig.get_config_var("HOST_GNU_TYPE") or "")
 REF_COUNT = 2
-CHAINED_WARNING_DISABLED = PYPY or (PY314 and not sys._is_gil_enabled())  # type: ignore[attr-defined]
-CHAINED_WARNING_DISABLED_INPLACE_METHOD = PYPY or PY314
+WARNING_CHECK_DISABLED = PY314
 
 
 __all__ = [

pandas/compat/pickle_compat.py

@@ -22,7 +22,6 @@
     PeriodArray,
     TimedeltaArray,
 )
-from pandas.core.generic import NDFrame
 from pandas.core.internals import BlockManager
 
 if TYPE_CHECKING:
@@ -91,10 +90,6 @@ def load_reduce(self) -> None:
                 cls = args[0]
                 stack[-1] = NDArrayBacked.__new__(*args)
                 return
-            elif args and issubclass(args[0], NDFrame):
-                cls = args[0]
-                stack[-1] = cls.__new__(cls)
-                return
             raise
 
     dispatch[pickle.REDUCE[0]] = load_reduce  # type: ignore[assignment]

pandas/core/frame.py

@@ -48,11 +48,11 @@
     properties,
 )
 from pandas._libs.hashtable import duplicated
-from pandas._libs.internals import SetitemMixin
 from pandas._libs.lib import is_range_indexer
+from pandas.compat import PYPY
 from pandas.compat._constants import (
-    CHAINED_WARNING_DISABLED_INPLACE_METHOD,
     REF_COUNT,
+    WARNING_CHECK_DISABLED,
 )
 from pandas.compat._optional import import_optional_dependency
 from pandas.compat.numpy import function as nv
@@ -63,6 +63,7 @@
 )
 from pandas.errors.cow import (
     _chained_assignment_method_msg,
+    _chained_assignment_msg,
 )
 from pandas.util._decorators import (
     Appender,
@@ -520,7 +521,7 @@
 
 
 @set_module("pandas")
-class DataFrame(SetitemMixin, NDFrame, OpsMixin):
+class DataFrame(NDFrame, OpsMixin):
     """
     Two-dimensional, size-mutable, potentially heterogeneous tabular data.
 
@@ -667,11 +668,6 @@ class DataFrame(SetitemMixin, NDFrame, OpsMixin):
     #  and ExtensionArray.  Should NOT be overridden by subclasses.
     __pandas_priority__ = 4000
 
-    # override those to avoid inheriting from SetitemMixin (cython generates
-    # them by default)
-    __reduce__ = object.__reduce__
-    __setstate__ = NDFrame.__setstate__
-
     @property
     def _constructor(self) -> type[DataFrame]:
         return DataFrame
@@ -4325,8 +4321,7 @@ def isetitem(self, loc, value) -> None:
         arraylike, refs = self._sanitize_column(value)
         self._iset_item_mgr(loc, arraylike, inplace=False, refs=refs)
 
-    # def __setitem__() is implemented in SetitemMixin and dispatches to this method
-    def _setitem(self, key, value) -> None:
+    def __setitem__(self, key, value) -> None:
         """
         Set item(s) in DataFrame by key.
 
@@ -4410,6 +4405,12 @@ def _setitem(self, key, value) -> None:
         z  3  50
         # Values for 'a' and 'b' are completely ignored!
         """
+        if not PYPY and not WARNING_CHECK_DISABLED:
+            if sys.getrefcount(self) <= REF_COUNT + 1:
+                warnings.warn(
+                    _chained_assignment_msg, ChainedAssignmentError, stacklevel=2
+                )
+
         key = com.apply_if_callable(key, self)
 
         # see if we can slice the rows
@@ -9362,7 +9363,7 @@ def update(
         1  2  500.0
         2  3    6.0
         """
-        if not CHAINED_WARNING_DISABLED_INPLACE_METHOD:
+        if not PYPY and not WARNING_CHECK_DISABLED:
             if sys.getrefcount(self) <= REF_COUNT:
                 warnings.warn(
                     _chained_assignment_method_msg,

pandas/core/generic.py

@@ -82,9 +82,10 @@
     WriteExcelBuffer,
     npt,
 )
+from pandas.compat import PYPY
 from pandas.compat._constants import (
-    CHAINED_WARNING_DISABLED_INPLACE_METHOD,
     REF_COUNT,
+    WARNING_CHECK_DISABLED,
 )
 from pandas.compat._optional import import_optional_dependency
 from pandas.compat.numpy import function as nv
@@ -2074,6 +2075,7 @@ def __getstate__(self) -> dict[str, Any]:
             **meta,
         }
 
+    @final
     def __setstate__(self, state) -> None:
         if isinstance(state, BlockManager):
             self._mgr = state
@@ -4264,9 +4266,8 @@ def _slice(self, slobj: slice, axis: AxisInt = 0) -> Self:
         result = result.__finalize__(self)
         return result
 
-    # def __delitem__() is implemented in SetitemMixin and dispatches to this method
     @final
-    def _delitem(self, key) -> None:
+    def __delitem__(self, key) -> None:
         """
         Delete item
         """
@@ -7081,7 +7082,7 @@ def fillna(
         """
         inplace = validate_bool_kwarg(inplace, "inplace")
         if inplace:
-            if not CHAINED_WARNING_DISABLED_INPLACE_METHOD:
+            if not PYPY and not WARNING_CHECK_DISABLED:
                 if sys.getrefcount(self) <= REF_COUNT:
                     warnings.warn(
                         _chained_assignment_method_msg,
@@ -7328,7 +7329,7 @@ def ffill(
         """
         inplace = validate_bool_kwarg(inplace, "inplace")
         if inplace:
-            if not CHAINED_WARNING_DISABLED_INPLACE_METHOD:
+            if not PYPY and not WARNING_CHECK_DISABLED:
                 if sys.getrefcount(self) <= REF_COUNT:
                     warnings.warn(
                         _chained_assignment_method_msg,
@@ -7468,7 +7469,7 @@ def bfill(
         """
         inplace = validate_bool_kwarg(inplace, "inplace")
         if inplace:
-            if not CHAINED_WARNING_DISABLED_INPLACE_METHOD:
+            if not PYPY and not WARNING_CHECK_DISABLED:
                 if sys.getrefcount(self) <= REF_COUNT:
                     warnings.warn(
                         _chained_assignment_method_msg,
@@ -7553,7 +7554,7 @@ def replace(
 
         inplace = validate_bool_kwarg(inplace, "inplace")
         if inplace:
-            if not CHAINED_WARNING_DISABLED_INPLACE_METHOD:
+            if not PYPY and not WARNING_CHECK_DISABLED:
                 if sys.getrefcount(self) <= REF_COUNT:
                     warnings.warn(
                         _chained_assignment_method_msg,
@@ -7916,7 +7917,7 @@ def interpolate(
         inplace = validate_bool_kwarg(inplace, "inplace")
 
         if inplace:
-            if not CHAINED_WARNING_DISABLED_INPLACE_METHOD:
+            if not PYPY and not WARNING_CHECK_DISABLED:
                 if sys.getrefcount(self) <= REF_COUNT:
                     warnings.warn(
                         _chained_assignment_method_msg,
@@ -8571,7 +8572,7 @@ def clip(
         inplace = validate_bool_kwarg(inplace, "inplace")
 
         if inplace:
-            if not CHAINED_WARNING_DISABLED_INPLACE_METHOD:
+            if not PYPY and not WARNING_CHECK_DISABLED:
                 if sys.getrefcount(self) <= REF_COUNT:
                     warnings.warn(
                         _chained_assignment_method_msg,
@@ -10206,7 +10207,7 @@ def where(
         """
         inplace = validate_bool_kwarg(inplace, "inplace")
         if inplace:
-            if not CHAINED_WARNING_DISABLED_INPLACE_METHOD:
+            if not PYPY and not WARNING_CHECK_DISABLED:
                 if sys.getrefcount(self) <= REF_COUNT:
                     warnings.warn(
                         _chained_assignment_method_msg,
@@ -10270,7 +10271,7 @@ def mask(
     ) -> Self | None:
         inplace = validate_bool_kwarg(inplace, "inplace")
         if inplace:
-            if not CHAINED_WARNING_DISABLED_INPLACE_METHOD:
+            if not PYPY and not WARNING_CHECK_DISABLED:
                 if sys.getrefcount(self) <= REF_COUNT:
                     warnings.warn(
                         _chained_assignment_method_msg,

pandas/core/indexing.py

@@ -15,9 +15,10 @@
 
 from pandas._libs.indexing import NDFrameIndexerBase
 from pandas._libs.lib import item_from_zerodim
+from pandas.compat import PYPY
 from pandas.compat._constants import (
-    CHAINED_WARNING_DISABLED,
     REF_COUNT,
+    WARNING_CHECK_DISABLED,
 )
 from pandas.errors import (
     AbstractMethodError,
@@ -919,7 +920,7 @@ def _ensure_listlike_indexer(self, key, axis=None, value=None) -> None:
 
     @final
     def __setitem__(self, key, value) -> None:
-        if not CHAINED_WARNING_DISABLED:
+        if not PYPY and not WARNING_CHECK_DISABLED:
             if sys.getrefcount(self.obj) <= REF_COUNT:
                 warnings.warn(
                     _chained_assignment_msg, ChainedAssignmentError, stacklevel=2
@@ -2587,7 +2588,7 @@ def __getitem__(self, key):
         return super().__getitem__(key)
 
     def __setitem__(self, key, value) -> None:
-        if not CHAINED_WARNING_DISABLED:
+        if not PYPY and not WARNING_CHECK_DISABLED:
             if sys.getrefcount(self.obj) <= REF_COUNT:
                 warnings.warn(
                     _chained_assignment_msg, ChainedAssignmentError, stacklevel=2
@@ -2618,7 +2619,7 @@ def _convert_key(self, key):
         return key
 
     def __setitem__(self, key, value) -> None:
-        if not CHAINED_WARNING_DISABLED:
+        if not PYPY and not WARNING_CHECK_DISABLED:
             if sys.getrefcount(self.obj) <= REF_COUNT:
                 warnings.warn(
                     _chained_assignment_msg, ChainedAssignmentError, stacklevel=2