Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix unjitted drop food function in aquarium step #6

Merged
merged 1 commit into from
Jul 16, 2024

Conversation

corentinlger
Copy link
Owner

  • replace the idx sampling of new food method in the drop_food() fn
  • use jax.lax.cond instead of if statement to call drop_food()

@corentinlger corentinlger merged commit 15f2e8d into main Jul 16, 2024
@corentinlger corentinlger deleted the fix_cond_aquarium_step branch July 16, 2024 13:59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant