Commit c8f2280f authored by Carl Goodrich's avatar Carl Goodrich
Browse files

change jnp.int16 to int

parent c3a52a0d
......@@ -102,7 +102,7 @@ def vector2symmat(v):
1, 3, 4],
2, 4, 5]]
"""
n = jnp.int16(((1 + 8 * v.shape[0]) ** 0.5 - 1) / 2)
n = int(((1 + 8 * v.shape[0]) ** 0.5 - 1) / 2)
return _vector2symmat(v, jnp.zeros((n, n), dtype=v.dtype))
......@@ -131,5 +131,5 @@ def vector2symmat_diag0(v):
0, 0, 2],
1, 2, 0]]
"""
n = jnp.int16(((1 + 8 * v.shape[0]) ** 0.5 + 1) / 2)
n = int(((1 + 8 * v.shape[0]) ** 0.5 + 1) / 2)
return _vector2symmat_diag0(v, jnp.zeros((n, n), dtype=v.dtype))
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment