rxnn 0.2.14__tar.gz → 0.2.15__tar.gz

{rxnn-0.2.14 → rxnn-0.2.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rxnn
-Version: 0.2.14
+Version: 0.2.15
 Summary: RxNN: Reactive Neural Networks Platform
 License: Apache-2.0
 Keywords: deep-learning,ai,machine-learning

{rxnn-0.2.14 → rxnn-0.2.15}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "rxnn"
-version = "0.2.14"
+version = "0.2.15"
 description = "RxNN: Reactive Neural Networks Platform"
 
 license = "Apache-2.0"

{rxnn-0.2.14 → rxnn-0.2.15}/src/rxnn/training/dataset.py

@@ -977,6 +977,7 @@ class MrlCurriculumDataset(Dataset):
             answer_field: str = 'answer',
             interactions_field: str = 'interactions',
             load_kwargs: dict = None,
+            max_seq_len: int = 1024,
             **kwargs
     ):
         """
@@ -993,6 +994,7 @@ class MrlCurriculumDataset(Dataset):
             answer_field (str): Answer field (default: "answer")
             interactions_field (str): Interactions field (default: "interactions")
             load_kwargs (dict): Additional args for HuggingFace API load_dataset function
+            max_seq_len (int): Maximum sequence length (default: 1024)
             **kwargs: Additional args for RxNN Dataset class
         """
         if load_kwargs is None:
@@ -1000,7 +1002,7 @@ class MrlCurriculumDataset(Dataset):
 
         hf_dataset = load_dataset(dataset_id, mrl_subset, split=split, **load_kwargs)
 
-        return cls(hf_dataset, tokenizer, query_field=query_field, answer_field=answer_field, interactions_field=interactions_field, **kwargs)
+        return cls(hf_dataset, tokenizer, query_field=query_field, answer_field=answer_field, interactions_field=interactions_field, max_seq_len=max_seq_len, **kwargs)
 
     @staticmethod
     def collate_mrl_batch(batch: list[MrlDataItem]) -> MrlDataItem: