Skip to content

Commit

Permalink
fix: no round method
Browse files Browse the repository at this point in the history
  • Loading branch information
beckermr committed Sep 9, 2024
1 parent 565e938 commit 15d8edd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion jax_galsim/position.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,8 +134,8 @@ def shear(self, shear):
shear_pos = jnp.dot(shear_mat, self._array)
return PositionD(shear_pos[0], shear_pos[1])

@implements(_galsim.Position.round)
def round(self):
"""Return the rounded-off PositionI version of this position."""
return PositionI(jnp.round(self.x), jnp.round(self.y))

def tree_flatten(self):
Expand Down

0 comments on commit 15d8edd

Please sign in to comment.