Fix/api references (#4)

* Fix defines.py in __init__

* Fix typing

* Fix bug in the function set_kernel_cache_directory

Author: roaffix

arrayfire_wrapper/__init__.py

@@ -7,10 +7,10 @@
 __all__ += ["__arrayfire_version__"]
 __arrayfire_version__ = ARRAYFIRE_VERSION
 
-__all__ += ["AFArray"]
-from .defines import AFArray
+__all__ += ["AFArray", "ArrayBuffer", "CType"]
+from .defines import AFArray, ArrayBuffer, CType
 
-__all__ = [
+__all__ += [
     "Dtype",
     "b8",
     "bool",

arrayfire_wrapper/lib/__init__.py

@@ -8,7 +8,7 @@
 
 __all__ += ["CIndexStructure", "IndexStructure", "ParallelRange", "get_indices"]
 
-from .create_and_modify_array.assignment_and_indexing.indexing import (
+from .create_and_modify_array.assignment_and_indexing._indexing import (
     CIndexStructure,
     IndexStructure,
     ParallelRange,

arrayfire_wrapper/lib/create_and_modify_array/__init__.py

@@ -10,7 +10,7 @@
 
 __all__ += ["CIndexStructure", "IndexStructure", "ParallelRange", "get_indices"]
 
-from .assignment_and_indexing.indexing import CIndexStructure, IndexStructure, ParallelRange, get_indices
+from .assignment_and_indexing._indexing import CIndexStructure, IndexStructure, ParallelRange, get_indices
 
 __all__ += [
     "AFRandomEngineHandle",

…rray/assignment_and_indexing/indexing.py …ray/assignment_and_indexing/_indexing.pyarrayfire_wrapper/lib/create_and_modify_array/assignment_and_indexing/indexing.py renamed to arrayfire_wrapper/lib/create_and_modify_array/assignment_and_indexing/_indexing.py arrayfire_wrapper/lib/create_and_modify_array/assignment_and_indexing/indexing.py renamed to arrayfire_wrapper/lib/create_and_modify_array/assignment_and_indexing/_indexing.py

@@ -0,0 +1,18 @@
+import ctypes
+from typing import Any
+
+from arrayfire_wrapper._backend import _backend
+from arrayfire_wrapper.defines import AFArray, CDimT
+
+from ..._error_handler import safe_call
+
+# TODO unfinished module
+
+
+def index_gen(arr: AFArray, ndims: int, indices: Any, /) -> AFArray:
+    """
+    source: https://arrayfire.org/docs/group__index__func__index.htm#ga14a7d149dba0ed0b977335a3df9d91e6
+    """
+    out = AFArray.create_null_pointer()
+    safe_call(_backend.clib.af_index_gen(ctypes.pointer(out), arr, CDimT(ndims), indices.pointer))
+    return out

arrayfire_wrapper/lib/create_and_modify_array/assignment_and_indexing/index.py

@@ -14,7 +14,7 @@ def alloc_host(num_bytes: int, /) -> int:
     """
     out = AFArray.create_null_pointer()
     safe_call(_backend.clib.af_alloc_host(ctypes.pointer(out), CDimT(num_bytes)))
-    return out.value
+    return out.value  # type: ignore[return-value]
 
 
 def alloc_device(num_bytes: int, /) -> int:
@@ -26,7 +26,7 @@ def alloc_device(num_bytes: int, /) -> int:
     """
     out = AFArray.create_null_pointer()
     safe_call(_backend.clib.af_alloc_device_v2(ctypes.pointer(out), CDimT(num_bytes)))
-    return out.value
+    return out.value  # type: ignore[return-value]
 
 
 def device_info() -> dict[str, str]:
@@ -131,7 +131,7 @@ def get_device_ptr(arr: AFArray) -> int:
     """
     out = AFArray.create_null_pointer()
     safe_call(_backend.clib.af_get_device_ptr(ctypes.pointer(out), arr))
-    return out.value
+    return out.value  # type: ignore[return-value]
 
 
 def get_kernel_cache_directory(length: int, path: str, /) -> None:
@@ -246,7 +246,7 @@ def set_kernel_cache_directory(path: str, override_env: int, /) -> None:
     """
     source: https://arrayfire.org/docs/group__device__func__mem.htm#ga880be5cb0035d4f173d074ad06bce6a7
     """
-    safe_call(_backend.clib.af_set_kernel_cache_directory(path.encode("utf-8", override_env)))
+    safe_call(_backend.clib.af_set_kernel_cache_directory(path.encode("utf-8"), override_env))
 
 
 def set_mem_step_size(step_bytes: int, /) -> None:
@@ -397,7 +397,7 @@ def alloc_pinned(num_bytes: int, /) -> int:
     """
     out = ctypes.c_void_p(0)
     safe_call(_backend.clib.af_alloc_pinned(ctypes.pointer(out), CDimT(num_bytes)))
-    return out.value
+    return out.value  # type: ignore[return-value]
 
 
 def get_device() -> int: