Fix nlargest and nsmallest Series support (modin-project#46) (core)

Signed-off-by: Naren Krishna <[email protected]>
vnlitvinov · Feb 13, 2023 · ba7ba0b · ba7ba0b
1 parent 00f126b
commit ba7ba0b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/modin/pandas/series.py b/modin/pandas/series.py
@@ -1290,9 +1290,9 @@ def nlargest(self, n=5, keep="first"):  # noqa: PR01, RT01, D200
             raise NotImplementedError(
                 "Series.nlargest is not implemented for empty Series."
             )
-        return DataFrame(
+        return Series(
             query_compiler=self._query_compiler.nlargest(
-                n, [self._query_compiler.columns[0]], keep
+                n=n, columns=self.name, keep=keep
             )
         )
 
@@ -1302,11 +1302,11 @@ def nsmallest(self, n=5, keep="first"):  # noqa: PR01, RT01, D200
         """
         if len(self._query_compiler.columns) == 0:
             raise NotImplementedError(
-                "Series.nlargest is not implemented for empty Series."
+                "Series.nsmallest is not implemented for empty Series."
             )
-        return DataFrame(
+        return Series(
             query_compiler=self._query_compiler.nsmallest(
-                n, [self._query_compiler.columns[0]], keep
+                n=n, columns=self.name, keep=keep
             )
         )