From fbd0d66b71ca251ead21dbe4c8927ea0acc4128f Mon Sep 17 00:00:00 2001
From: Levi Leal <leviplj@gmail.com>
Date: Tue, 19 Sep 2023 16:10:57 +0100
Subject: [PATCH] fix: chained_assignment could be warn, raise or None

https://github.com/pandas-dev/pandas/blob/1.5.x/pandas/core/config_init.py#L563-L575
---
 pandas-stubs/_config/config.pyi   | 2 +-
 pandas-stubs/core/config_init.pyi | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/pandas-stubs/_config/config.pyi b/pandas-stubs/_config/config.pyi
index 753371a08..c4260bdc1 100644
--- a/pandas-stubs/_config/config.pyi
+++ b/pandas-stubs/_config/config.pyi
@@ -118,7 +118,7 @@ class IO(DictWrapper):
     sql: IOSQL
 
 class Mode(DictWrapper):
-    chained_assignment: str
+    chained_assignment: Literal["warn", "raise"] | None
     data_manager: str
     sim_interactive: bool
     string_storage: str
diff --git a/pandas-stubs/core/config_init.pyi b/pandas-stubs/core/config_init.pyi
index 773f3f640..d3791303f 100644
--- a/pandas-stubs/core/config_init.pyi
+++ b/pandas-stubs/core/config_init.pyi
@@ -1,3 +1,5 @@
+from typing import Literal
+
 use_bottleneck_doc: str = ...
 
 def use_bottleneck_cb(key) -> None: ...
@@ -49,7 +51,7 @@ use_inf_as_na_doc: str = ...
 
 def use_inf_as_na_cb(key) -> None: ...
 
-chained_assignment: str = ...
+chained_assignment: Literal["warn", "raise"] | None
 reader_engine_doc: str = ...
 writer_engine_doc: str = ...
 parquet_engine_doc: str = ...