mask as float when parsing

pysersic · May 22, 2024 · 57847c8 · 57847c8
1 parent 5ce00e3
commit 57847c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysersic/pysersic.py b/pysersic/pysersic.py
@@ -668,7 +668,7 @@ def parse_mask(mask: ArrayLike = None, data: ArrayLike = None):
     if mask is None:
         return jnp.ones_like(data).astype(jnp.bool_)
     else:
-        return jnp.logical_not(jnp.array(mask.astype(int))).astype(jnp.bool_)
+        return jnp.logical_not(jnp.array(mask.astype(float))).astype(jnp.bool_)
 
 
 def check_input_data(