mypy

Author: andrewgsavage

pandas/core/arrays/base.py

@@ -2151,7 +2151,7 @@ def round(self, decimals: int = 0, *args, **kwargs) -> Self:
                 f"Cannot round {type(self)} dtype as it is non-numeric or boolean"
             )
         return self._from_sequence(
-            [round(element) if not isna(element) else element for element in self.data],
+            [round(element) if not isna(element) else element for element in self],
             dtype=self.dtype,
         )
 

pandas/tests/extension/test_numpy.py

@@ -270,10 +270,6 @@ def test_insert_invalid(self, data, invalid_scalar):
         # NumpyExtensionArray[object] can hold anything, so skip
         super().test_insert_invalid(data, invalid_scalar)
 
-    @pytest.mark.xfail(reason="NumpyExtensionArray.round is not implemented.")
-    def test_round(self, data):
-        super().test_round(data)
-
 
 class TestArithmetics(BaseNumPyTests, base.BaseArithmeticOpsTests):
     divmod_exc = None