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

Signed-off-by: Naren Krishna <[email protected]>
vnlitvinov · Mar 16, 2023 · 163cee5 · 163cee5
1 parent f9d9b5f
commit 163cee5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/modin/pandas/series.py b/modin/pandas/series.py
@@ -1327,9 +1327,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
             )
         )
 
@@ -1339,11 +1339,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 self.__constructor__(
             query_compiler=self._query_compiler.nsmallest(
-                n, [self._query_compiler.columns[0]], keep
+                n=n, columns=self.name, keep=keep
             )
         )