Skip to content

Clean pufferl.py, delete Neptune - #595

Draft
vcharraut wants to merge 5 commits into
3.0from
vcha/refactor-pufferl
Draft

vcharraut wants to merge 5 commits into
3.0from
vcha/refactor-pufferl

Conversation

@vcharraut

Copy link
Copy Markdown
Collaborator

No description provided.

- Added various utility functions in `pufferlib/utils.py` for device management, state handling, and logging.
- Implemented functions for cleaning policy state dictionaries, downsampling data, and capturing/restoring RNG states.
- Introduced `get_git_metadata` and `save_experiment_config` for better experiment tracking.
- Updated tests in `tests/eval/test_eval.py` to use public methods instead of private ones for better maintainability.
- Refactored rank seeding tests in `tests/unit_tests/test_rank_seeding.py` to utilize the new utility functions.

This branch has not been deployed

No deployments
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