fix: don't convert python's int

EMI-Group · Sep 5, 2023 · 921d61a · 921d61a
1 parent d9fb0f1
commit 921d61a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/evox/operators/sampling/uniform.py b/src/evox/operators/sampling/uniform.py
@@ -26,7 +26,7 @@ def __call__(self):
             jnp.array(list(n_choose_k(range(1, h1 + self.m), self.m - 1)))
             - jnp.tile(
                 jnp.array(range(self.m - 1)),
-                (comb(h1 + self.m - 1, self.m - 1).astype(int), 1),
+                (comb(h1 + self.m - 1, self.m - 1), 1),
             )
             - 1
         )
@@ -46,7 +46,7 @@ def __call__(self):
                     jnp.array(list(n_choose_k(range(1, h2 + self.m), self.m - 1)))
                     - jnp.tile(
                         jnp.array(range(self.m - 1)),
-                        (comb(h2 + self.m - 1, self.m - 1).astype(int), 1),
+                        (comb(h2 + self.m - 1, self.m - 1), 1),
                     )
                     - 1
                 )