From f5f9833540c07248c3f5a469c45408fe17843222 Mon Sep 17 00:00:00 2001 From: Bryson Jones Date: Thu, 25 Dec 2025 04:14:56 -0800 Subject: [PATCH] add kwargs arg to multitask dit constructor --- src/lerobot/policies/multi_task_dit/modeling_multi_task_dit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/policies/multi_task_dit/modeling_multi_task_dit.py b/src/lerobot/policies/multi_task_dit/modeling_multi_task_dit.py index 0d252b296..4fee851e0 100644 --- a/src/lerobot/policies/multi_task_dit/modeling_multi_task_dit.py +++ b/src/lerobot/policies/multi_task_dit/modeling_multi_task_dit.py @@ -64,7 +64,7 @@ class MultiTaskDiTPolicy(PreTrainedPolicy): config_class = MultiTaskDiTConfig name = "multi_task_dit" - def __init__(self, config: MultiTaskDiTConfig): + def __init__(self, config: MultiTaskDiTConfig, **kwargs): super().__init__(config) config.validate_features() self.config = config