[Refactor] Re-export History from tensordict.llm

test/llm/test_data.py

@@ -29,6 +29,7 @@
     _CUSTOM_MODEL_FAMILY_KEYWORDS,
     add_chat_template,
     ContentBase,
+    history_default_spec,
 )
 from torchrl.data.llm.topk import TopKRewardSelector
 from torchrl.modules.llm.policies.common import _extract_responses_from_full_histories
@@ -300,6 +301,11 @@ def test_history_spec(self):
         assert isinstance(r, History)
         assert spec.is_in(r)
         assert spec.is_in(history)
+        # The free function is the canonical API now that History lives in
+        # tensordict; the classmethod above is kept for backward compatibility.
+        spec_fn = history_default_spec()
+        assert spec_fn.is_in(history)
+        assert spec_fn.is_in(r)
 
     def test_content_base(self):
         from transformers import AutoProcessor

torchrl/data/llm/__init__.py

@@ -9,7 +9,7 @@
     TensorDictTokenizer,
     TokenizedDatasetLoader,
 )
-from .history import add_chat_template, ContentBase, History
+from .history import add_chat_template, ContentBase, History, history_default_spec
 from .prompt import PromptData, PromptTensorDictTokenizer
 from .reward import PairwiseDataset, RewardData
 from .topk import TopKRewardSelector
@@ -30,5 +30,6 @@
     "TokenizedDatasetLoader",
     "create_infinite_iterator",
     "get_dataloader",
+    "history_default_spec",
     "TopKRewardSelector",
 ]