Set array namespace in _create_from_usm_ndarray

Author: antonwolfy

dpnp/dpnp_array.py

@@ -353,15 +353,7 @@ def __getitem__(self, key):
         key = _get_unwrapped_index_key(key)
 
         item = self._array_obj.__getitem__(key)
-        if not isinstance(item, dpt.usm_ndarray):
-            raise RuntimeError(
-                "Expected dpctl.tensor.usm_ndarray, got {}"
-                "".format(type(item))
-            )
-
-        res = self.__new__(dpnp_array)
-        res._array_obj = item
-        return res
+        return dpnp._create_from_usm_ndarray(item)
 
     # '__getstate__',
 
@@ -632,6 +624,7 @@ def _create_from_usm_ndarray(usm_ary: dpt.usm_ndarray):
             )
         res = dpnp_array.__new__(dpnp_array)
         res._array_obj = usm_ary
+        res._array_obj._set_namespace(dpnp)
         return res
 
     def all(self, axis=None, out=None, keepdims=False, *, where=True):
@@ -1775,17 +1768,16 @@ def transpose(self, *axes):
         if axes_len == 1 and isinstance(axes[0], (tuple, list)):
             axes = axes[0]
 
-        res = self.__new__(dpnp_array)
         if ndim == 2 and axes_len == 0:
-            res._array_obj = self._array_obj.T
+            usm_res = self._array_obj.T
         else:
             if len(axes) == 0 or axes[0] is None:
                 # self.transpose().shape == self.shape[::-1]
                 # self.transpose(None).shape == self.shape[::-1]
                 axes = tuple((ndim - x - 1) for x in range(ndim))
 
-            res._array_obj = dpt.permute_dims(self._array_obj, axes)
-        return res
+            usm_res = dpt.permute_dims(self._array_obj, axes)
+        return dpnp._create_from_usm_ndarray(usm_res)
 
     def var(
         self,