Skip to content

Commit

Permalink
Save TorchScripted model for LinearBandit
Browse files Browse the repository at this point in the history
Summary: Change Pearl FBLearner Flow workflow to save TorchScripted model for `LinearBandit`.

Reviewed By: PavlosApo

Differential Revision: D65865605

fbshipit-source-id: a33fceeaf85382c3880318a2b6342491b3180399
  • Loading branch information
rodrigodesalvobraz authored and facebook-github-bot committed Nov 15, 2024
1 parent 9c8db81 commit 357ca0c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pearl/replay_buffers/tensor_based_replay_buffer.py
Original file line number Diff line number Diff line change
Expand Up @@ -262,7 +262,7 @@ def sample(self, batch_size: int) -> TransitionBatch:
"""
if batch_size > len(self):
raise ValueError(
f"Can't get a batch of size {batch_size} from a replay buffer with"
f"Can't get a batch of size {batch_size} from a replay buffer with "
f"only {len(self)} elements"
)
samples = random.sample(self.memory, batch_size)
Expand Down

0 comments on commit 357ca0c

Please sign in to comment.