fix: binding types

abetlen · Feb 9, 2024 · 4fed8db · 4fed8db
1 parent cb98ede
commit 4fed8db
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ggml/ggml.py b/ggml/ggml.py
@@ -1320,7 +1320,7 @@ def ggml_tensor_overhead() -> int:
 # GGML_API struct ggml_context * ggml_init(struct ggml_init_params params);
 def ggml_init(
     params: ggml_init_params,
-) -> ggml_context_p:
+) -> Optional[ggml_context_p]:
     """Instantiate a new ggml context with params.
 
     You must call `ggml_free()` to free the context.
@@ -1329,7 +1329,7 @@ def ggml_init(
         params: ggml init params
 
     Returns:
-        Pointer to ggml_context"""
+        Pointer to ggml_context or None if failed to initialize context."""
     return lib.ggml_init(params)
 
 
@@ -9766,8 +9766,8 @@ def ggml_backend_reg_alloc_buffer(
 # // Initialize a backend scheduler
 # GGML_API ggml_backend_sched_t  ggml_backend_sched_new(ggml_backend_t * backends, ggml_backend_buffer_type_t * bufts, int n_backends, size_t graph_size);
 def ggml_backend_sched_new(
-    backends: "ctypes._Pointer(ggml_backend_t)",  # type: ignore
-    bufts: "ctypes._Pointer(ggml_backend_buffer_type_t)",  # type: ignore
+    backends: "ctypes._Pointer[ggml_backend_t]",  # type: ignore
+    bufts: "ctypes._Pointer[ggml_backend_buffer_type_t]",  # type: ignore
     n_backends: Union[ctypes.c_int, int],
     graph_size: Union[ctypes.c_size_t, int],
 ) -> ggml_backend_sched_t: