Skip to content

[Feature] Added Lazy implementation of priority updates for replaybuffer prototype#3507

Open
ParamThakkar123 wants to merge 23 commits intopytorch:mainfrom
ParamThakkar123:add/lazy
Open

[Feature] Added Lazy implementation of priority updates for replaybuffer prototype#3507
ParamThakkar123 wants to merge 23 commits intopytorch:mainfrom
ParamThakkar123:add/lazy

Conversation

@ParamThakkar123
Copy link
Contributor

Description

Describe your changes in detail.

Motivation and Context

Why is this change required? What problem does it solve?
If it fixes an open issue, please link to the issue here.
You can use the syntax close #15213 if this solves the issue #15213

Fixes #616

  • I have raised an issue to propose this change (required for new features and bug fixes)

Types of changes

What types of changes does your code introduce? Remove all that do not apply:

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds core functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)
  • Documentation (update in the documentation)
  • Example (update in the folder of examples)

Checklist

Go over all the following points, and put an x in all the boxes that apply.
If you are unsure about any of these, don't hesitate to ask. We are here to help!

  • I have read the CONTRIBUTION guide (required)
  • My change requires a change to the documentation.
  • I have updated the tests accordingly (required for a bug fix or a new feature).
  • I have updated the documentation accordingly.

@pytorch-bot
Copy link

pytorch-bot bot commented Feb 13, 2026

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/rl/3507

Note: Links to docs will display an error until the docs builds have been completed.

⚠️ 17 Awaiting Approval

As of commit 62cde31 with merge base 83c2101 (image):

AWAITING APPROVAL - The following workflows need approval before CI can run:

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@meta-cla meta-cla 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 Feb 13, 2026
vmoens and others added 3 commits February 13, 2026 16:18
- Remove dead _is_int check in _update_priority_tree (always False after
  _normalize_index which guarantees ndim >= 1).
- Avoid double index normalization on the flush path: _update_priority_tree
  now expects pre-normalized, pre-filtered inputs; callers (update_priority,
  mark_update, _flush_pending_updates) handle normalization and negative-index
  filtering themselves.
- Add docstring to mark_update explaining the lazy semantics.
- Add tests for multiple-extend accumulation, empty-clears-pending, and
  state_dict round-trip with pending updates.

Co-authored-by: Cursor <cursoragent@cursor.com>
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. Feature New feature ReplayBuffers

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[Feature Request] lazy implementation of priority updates for prototype ReplayBuffer

2 participants