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

small fixes to seeding and hypergrid tests, docstring improvements to resolve confusion of get_terminating_states_indices purpose #217

Merged
merged 3 commits into from
Nov 15, 2024

Conversation

josephdviviano
Copy link
Collaborator

@josephdviviano josephdviviano commented Nov 15, 2024

see title

was inspired by this issue #205

@josephdviviano josephdviviano added bug Something isn't working small Small enhancement labels Nov 15, 2024
@josephdviviano josephdviviano self-assigned this Nov 15, 2024
@josephdviviano josephdviviano changed the title terminating_states are now calculated using self.all_states small fixes to seeding and hypergrid tests, docstring improvements to resolve confusion of get_terminating_states_indices purpose Nov 15, 2024
Copy link
Collaborator

@hyeok9855 hyeok9855 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@@ -86,6 +86,7 @@ def update_masks(self, states: type[DiscreteStates]) -> None:
"""Update the masks based on the current states."""
# Not allowed to take any action beyond the environment height, but
# allow early termination.
# TODO: do we need to handle the conditional case here?
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I don't think so. We may need another EnvBase that incorporates conditions.

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ConditionalEnvs are interesting. I should actually do this ASAP so we have a working conditional GFN example.

@josephdviviano josephdviviano merged commit 6f132a8 into master Nov 15, 2024
4 checks passed
@josephdviviano josephdviviano deleted the remove_unused_method branch November 15, 2024 14:48
@josephdviviano josephdviviano restored the remove_unused_method branch November 19, 2024 18:01
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working small Small enhancement
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants