torchrl-nightly 2025.7.17__cp310-cp310-win_amd64.whl → 2025.7.18__cp310-cp310-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Binary file
@@ -713,6 +713,42 @@ class History(TensorClass["nocast"]):
713
713
  | transformers.AutoProcessor # noqa: F821
714
714
  | None = None,
715
715
  ) -> History:
716
+ r"""Inverts a chat template into a History object.
717
+
718
+ Args:
719
+ text (str | list[str]): The chat template to invert.
720
+ chat_template_name (str, optional): The name of the chat template to use.
721
+ tokenizer (transformers.AutoTokenizer | transformers.AutoProcessor, optional): The tokenizer to use.
722
+
723
+ Returns:
724
+ History: The inverted History object.
725
+
726
+ Examples:
727
+ >>> from torchrl.data.llm.history import History
728
+ >>> from transformers import AutoTokenizer
729
+ >>> tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-7B-Instruct")
730
+ >>> text = "<|im_start|>system\nYou are a helpful assistant.\n<|im_end|>\n<|im_start|>user\nWrite a python script that gives the capital of France or Germany.\n<|im_end|>\n<|im_start|>assistant\n<think>The capital of France is Paris, the capital of Germany is Berlin.</think>\n<answer><python>\n"
731
+ >>> history = History.from_text(text, tokenizer=tokenizer)
732
+ >>> print(history)
733
+ History(
734
+ content=NonTensorStack(
735
+ ['You are a helpful assistant.', 'Write a python s...,
736
+ batch_size=torch.Size([3]),
737
+ device=None),
738
+ is_complete=NonTensorStack(
739
+ [True, True, False],
740
+ batch_size=torch.Size([3]),
741
+ device=None),
742
+ role=NonTensorStack(
743
+ ['system', 'user', 'assistant'],
744
+ batch_size=torch.Size([3]),
745
+ device=None),
746
+ tool_calls=None,
747
+ tool_responses=None,
748
+ batch_size=torch.Size([3]),
749
+ device=None,
750
+ is_shared=False)
751
+ """
716
752
  if chat_template_name is None:
717
753
  if chat_template is not None:
718
754
  # TODO: find best match given template
torchrl/version.py CHANGED
@@ -1,2 +1,2 @@
1
- __version__ = '2025.7.17'
2
- git_version = 'bec8f0382b9694a87c04385f55fc9f8f3ee1724f'
1
+ __version__ = '2025.7.18'
2
+ git_version = '4001d9cb73cea4498b0fdfe420effc58a5a336be'
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: torchrl-nightly
3
- Version: 2025.7.17
3
+ Version: 2025.7.18
4
4
  Summary: UNKNOWN
5
5
  Home-page: https://github.com/pytorch/rl
6
6
  Author: torchrl contributors
@@ -3,9 +3,9 @@ build_tools/setup_helpers/__init__.py,sha256=l9zlK7Nm5bT7P_onQx-hZeIGzKKyCFm1PFk
3
3
  build_tools/setup_helpers/extension.py,sha256=ihV8jz8kqOvpqzuD006XqF1oNX5ukKGlwIOJRb1Vd-o,6075
4
4
  torchrl/__init__.py,sha256=76lKYwYKmAKORhyVt2tURmYAIRTifxxO3gWsskrHAXU,3054
5
5
  torchrl/_extension.py,sha256=x6Nqj2brF3VhlEwxmNA2fYbmpxq1HHGrHMnP0YnQwdc,2412
6
- torchrl/_torchrl.cp310-win_amd64.pyd,sha256=EmXEHv8OHR8lomhJKYRoDdRNcjSZrLKyQdUKYT1-BR4,440832
6
+ torchrl/_torchrl.cp310-win_amd64.pyd,sha256=mX38cPRx8qM53-6r4E9Cvm0dGQ4UNdE_TnRTV6VU0Zg,440832
7
7
  torchrl/_utils.py,sha256=2N35rdD65U1khMi5gVIz8-nMjlZsoVq0kCiQftVRSxw,42297
8
- torchrl/version.py,sha256=hd0Oai-1iVD9JdvzREmm-B6pS3vtNIorROtxZuyZf7A,85
8
+ torchrl/version.py,sha256=RHg2XUmN6ktJoL3cFIF-B7Q9KTnIxzSg8yN0KerUchc,85
9
9
  torchrl/collectors/__init__.py,sha256=LzTyfxmkNGPSa5-3rS5unQK7HfT5ZEdr2NV291rAOlU,832
10
10
  torchrl/collectors/collectors.py,sha256=UbXtDMHrXVQ-cd95TBpss2SIbNFKze2HTPYjFz2cPNQ,182146
11
11
  torchrl/collectors/utils.py,sha256=aBmBLpphhfplqQjRCyn1jtWWJ-Wtc7TWvM0rOBN8SsE,11579
@@ -41,7 +41,7 @@ torchrl/data/datasets/utils.py,sha256=tRZkarWl-BX_fnGEVNRt6Fjo0wmyKCNAuVmw5Le-0C
41
41
  torchrl/data/datasets/vd4rl.py,sha256=YFjXvP-QGNzF7UWNKkGMKPFthcB0I8v6sJc7oESegYA,18694
42
42
  torchrl/data/llm/__init__.py,sha256=-_UPiaQgHzFVLM_gfiOS0sCGVRwkxZ1_Z5B6C9ThN9o,1009
43
43
  torchrl/data/llm/dataset.py,sha256=GDuzflBq2ThgYn_V4bOr_1MHOhESnQ5jX1Wlcw69lfM,21194
44
- torchrl/data/llm/history.py,sha256=UHWqvnEXejFDoz59XSAchVddLzkUS9uykYECDaDpX9I,59202
44
+ torchrl/data/llm/history.py,sha256=unGNaAklSwt7E1fQysN2Kpv9dcnuOmfCLwbX45pBgmw,61021
45
45
  torchrl/data/llm/prompt.py,sha256=ikHWafhTIoCONCpuMHwIuGfpnPSpg5drZQTxTCmygQE,8578
46
46
  torchrl/data/llm/reward.py,sha256=QW1HWpNRORd3InwWLg-hAhjTlPqX4ffzAkYHEz0jQxo,8629
47
47
  torchrl/data/llm/topk.py,sha256=E5rrMw_dXBpFsV_6DOwVSd714G6IKwjIxBZLpssmWLw,8551
@@ -223,8 +223,8 @@ torchrl/trainers/helpers/losses.py,sha256=HwrovwbMOhY-5-hlOz-YHclKnoJhMijVjDNuAT
223
223
  torchrl/trainers/helpers/models.py,sha256=VujBq9H92sEzpCtU1iTrJQNlwvyOO-Rho4bzsMonX6s,22465
224
224
  torchrl/trainers/helpers/replay_buffer.py,sha256=RaZqXnHimmadiibvDBcLbtIhpPaVMTPhYMOBvX4v3CA,2060
225
225
  torchrl/trainers/helpers/trainers.py,sha256=hB1FtHtP-S0PBQ4LF6WPy37caaLpacyaLThj1BNl5Ho,12372
226
- torchrl_nightly-2025.7.17.dist-info/LICENSE,sha256=PGO-oZsq4EzhE1-WQS2xGiEF3UCVb9YawfQ09cIMV_8,1119
227
- torchrl_nightly-2025.7.17.dist-info/METADATA,sha256=SaETkAw3q6ZUIazIqAwcLeC4WvSoghlUNoSQlhQ4nXQ,44000
228
- torchrl_nightly-2025.7.17.dist-info/WHEEL,sha256=NVXpD7b4Gxps0cd2ds5rr5TG8W4ApEwx_i5J99qMZ5E,102
229
- torchrl_nightly-2025.7.17.dist-info/top_level.txt,sha256=JeTJ1jV7QJwLcUS1nr21aPn_wb-XlAZ9c-z_EH472JA,20
230
- torchrl_nightly-2025.7.17.dist-info/RECORD,,
226
+ torchrl_nightly-2025.7.18.dist-info/LICENSE,sha256=PGO-oZsq4EzhE1-WQS2xGiEF3UCVb9YawfQ09cIMV_8,1119
227
+ torchrl_nightly-2025.7.18.dist-info/METADATA,sha256=CFFy5N5auTFjmeoKB9YxwVlQSI0_q2X5ITbRPOXhFXA,44000
228
+ torchrl_nightly-2025.7.18.dist-info/WHEEL,sha256=NVXpD7b4Gxps0cd2ds5rr5TG8W4ApEwx_i5J99qMZ5E,102
229
+ torchrl_nightly-2025.7.18.dist-info/top_level.txt,sha256=JeTJ1jV7QJwLcUS1nr21aPn_wb-XlAZ9c-z_EH472JA,20
230
+ torchrl_nightly-2025.7.18.dist-info/RECORD,,