Typing: Narrow down types of argument keep for nsmallest and nlargest (…

…pandas-dev#52753) * Typing: Narrow down types of argument keep for nsmallest and nlargest * fixed formatting with black
vagechirkov · Apr 20, 2023 · 75435a7 · 75435a7
1 parent 065b3d0
commit 75435a7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pandas/core/series.py b/pandas/core/series.py
@@ -3901,7 +3901,9 @@ def nlargest(
         """
         return selectn.SelectNSeries(self, n=n, keep=keep).nlargest()
 
-    def nsmallest(self, n: int = 5, keep: str = "first") -> Series:
+    def nsmallest(
+        self, n: int = 5, keep: Literal["first", "last", "all"] = "first"
+    ) -> Series:
         """
         Return the smallest `n` elements.