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

[Feature] Lazy imports for implement_for during torchrl import #1646

Merged
merged 2 commits into from
Oct 24, 2023

Conversation

vmoens
Copy link
Contributor

@vmoens vmoens commented Oct 24, 2023

Description

Makes implement_for lazy: the import that checks the package version will only occur the first time the function is run.
This allows us to skip the gym imports during torchrl import, and speeds up the lib import itself.

@matteobettini

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Oct 24, 2023
@vmoens vmoens added quality code quality performance Performance issue or suggestion for improvement labels Oct 24, 2023
Copy link
Contributor

@matteobettini matteobettini left a comment

Choose a reason for hiding this comment

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

Amazing!

@vmoens vmoens merged commit a67b9fb into main Oct 24, 2023
@vmoens vmoens deleted the lazy_implement_for branch October 24, 2023 14:07
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. performance Performance issue or suggestion for improvement quality code quality
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants