revert abstraction

Author: nayyirahsan

.github/workflows/array_api.yml

@@ -57,4 +57,4 @@ jobs:
           # for hypothesis-driven test case generation
           # pytest $GITHUB_WORKSPACE/tools/pre_compile_ufuncs.py -s
           # only run a subset of the conformance tests to get started
-          pytest array_api_tests/meta/test_broadcasting.py array_api_tests/meta/test_equality_mapping.py array_api_tests/meta/test_signatures.py array_api_tests/meta/test_special_cases.py array_api_tests/test_constants.py array_api_tests/meta/test_utils.py array_api_tests/test_creation_functions.py::test_ones array_api_tests/test_creation_functions.py::test_ones_like array_api_tests/test_data_type_functions.py::test_result_type array_api_tests/test_array_object.py::test_scalar_casting array_api_tests/test_operators_and_elementwise_functions.py::test_sign array_api_tests/test_operators_and_elementwise_functions.py::test_round array_api_tests/test_operators_and_elementwise_functions.py::test_trunc array_api_tests/test_operators_and_elementwise_functions.py::test_ceil array_api_tests/test_operators_and_elementwise_functions.py::test_floor array_api_tests/test_operators_and_elementwise_functions.py::test_isnan array_api_tests/test_operators_and_elementwise_functions.py::test_equal array_api_tests/test_creation_functions.py::test_zeros array_api_tests/test_creation_functions.py::test_zeros_like array_api_tests/test_creation_functions.py::test_full_like
+          pytest array_api_tests/meta/test_broadcasting.py array_api_tests/meta/test_equality_mapping.py array_api_tests/meta/test_signatures.py array_api_tests/meta/test_special_cases.py array_api_tests/test_constants.py array_api_tests/meta/test_utils.py array_api_tests/test_creation_functions.py::test_ones array_api_tests/test_creation_functions.py::test_ones_like array_api_tests/test_data_type_functions.py::test_result_type array_api_tests/test_array_object.py::test_scalar_casting array_api_tests/test_operators_and_elementwise_functions.py::test_isnan array_api_tests/test_operators_and_elementwise_functions.py::test_equal array_api_tests/test_creation_functions.py::test_zeros array_api_tests/test_creation_functions.py::test_zeros_like array_api_tests/test_creation_functions.py::test_full_like

pykokkos/__init__.py

@@ -34,26 +34,12 @@
 from pykokkos.interface.views import astype
 
 import numpy as np
-from pykokkos.lib.pk_array import PKArray
 
 
-def _pk_func(np_func):
-    def wrapper(*args, **kwargs):
-        return PKArray(np_func(*args, **kwargs))
-
-    return wrapper
-
-
-isnan = _pk_func(np.isnan)
-isinf = _pk_func(np.isinf)
-isfinite = _pk_func(np.isfinite)
-equal = _pk_func(np.equal)
-sign = _pk_func(np.sign)
-round = _pk_func(np.round)
-trunc = _pk_func(np.trunc)
-ceil = _pk_func(np.ceil)
-floor = _pk_func(np.floor)
-
+isnan = np.isnan
+isinf = np.isinf
+isfinite = np.isfinite
+equal = np.equal
 
 runtime_singleton.runtime = Runtime()