Merge branch 'main' of github.com:jasciiz/evox into pr

src/evox/problems/numerical/maf.py

@@ -440,7 +440,7 @@ def evaluate(self, state, X):
 
     def pf(self, state):
         n = self.ref_num * self.m
-        temp = jnp.linspace(-1, 1, num=math.ceil(math.sqrt(n)))
+        temp = jnp.linspace(-1, 1, num=jnp.ceil(jnp.sqrt(n)).astype(int))
         y, x = jnp.meshgrid(temp, temp)
         x = x.ravel(order="F")
         y = y.ravel(order="F")