update to 0.1.3; minor fixes

Author: Aaron-Preus

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "deterministic_gaussian_sampling"
-version = "0.1.2"
+version = "0.1.3"
 description = "Python library for Localized Distribution (LCD)-based Gaussian sampling."
 readme = "README.md"
 requires-python = ">=3.8"

src/deterministic_gaussian_sampling/approximation/base_approximation.py

@@ -28,16 +28,9 @@ def _check_numpy_ndarray(
             return None
         if not isinstance(arr, numpy.ndarray):
             raise TypeError("Input must be a numpy array")
-        if (
-            arr.dtype != float
-            and arr.dtype != numpy.float16
-            and arr.dtype != numpy.float32
-            and arr.dtype != numpy.float64
-            and arr.dtype != numpy.float96
-            and arr.dtype != numpy.float128
-        ):
+        if not numpy.issubdtype(arr.dtype, numpy.floating) and arr.dtype != float:
             raise TypeError(
-                f"Input array must be of [float, numpy.float16, numpy.float32, numpy.float64, numpy.float96, numpy.float128], but got {arr.dtype}."
+                f"Input array must be of a floating type, but got {arr.dtype}."
             )
         if arr.shape != (L, N):
             row, cols = arr.shape

src/deterministic_gaussian_sampling/approximation/gaussian_to_dirac.py

@@ -42,9 +42,9 @@ def approximate_double(
         success = cdll.gm_to_dirac_short_double_approximate(
             self.gm_to_dirac_double,
             self._check_numpy_ndarray(covDiag, covDiag.shape[0], covDiag.shape[0]),
-            ctypes.c_int(L),
-            ctypes.c_int(N),
-            ctypes.c_int(100),
+            ctypes.c_size_t(L),
+            ctypes.c_size_t(N),
+            ctypes.c_size_t(100),
             self._check_numpy_ndarray(x, L, N),
             self._check_numpy_ndarray(wX, L, 1),
             ctypes.byref(minimizer_result),
@@ -72,9 +72,9 @@ def approximate_snd_double(
         minimizer_result = ctypes_wrapper.GslMinimizerResultCTypes()
         success = cdll.gm_to_dirac_short_standard_normal_deviation_double_approximate(
             self.gm_to_dirac_snd_double,
-            ctypes.c_int(L),
-            ctypes.c_int(N),
-            ctypes.c_int(100),
+            ctypes.c_size_t(L),
+            ctypes.c_size_t(N),
+            ctypes.c_size_t(100),
             self._check_numpy_ndarray(x, L, N),
             self._check_numpy_ndarray(wX, L, 1),
             ctypes.byref(minimizer_result),