cleanup

Author: okaris

llama_cpp/_ctypes_extensions.py

@@ -5,8 +5,6 @@
 import ctypes
 import functools
 import pathlib
-import logging
-import traceback
 
 from typing import (
     Any,
@@ -20,9 +18,6 @@
 )
 from typing_extensions import TypeAlias
 
-# Configure logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger("llama_cpp.binding")
 
 # Load the library
 def load_shared_library(lib_base_name: str, base_path: pathlib.Path):
@@ -115,21 +110,11 @@ def ctypes_function(
     ):
         def decorator(f: F) -> F:
             if enabled:
-                print(f"Setting up binding for C function: {name}")  # Print when binding is created
                 func = getattr(lib, name)
                 func.argtypes = argtypes
                 func.restype = restype
-
-                @functools.wraps(f)
-                def wrapper(*args, **kwargs):
-                    print(f">>> Calling {name} with args: {args}")  # Print right before C call
-                    sys.stdout.flush()  # Force flush to ensure we see the output
-                    result = func(*args, **kwargs)
-                    print(f"<<< {name} returned successfully")  # Print after successful return
-                    sys.stdout.flush()
-                    return result
-
-                return wrapper
+                functools.wraps(f)(func)
+                return func
             else:
                 return f
 
@@ -143,4 +128,4 @@ def _byref(obj: CtypesCData, offset: Optional[int] = None) -> CtypesRef[CtypesCD
     ...
 
 
-byref = _byref if TYPE_CHECKING else ctypes.byref
+byref = _byref if TYPE_CHECKING else ctypes.byref