From 68173f05df14ed98b79fd3f465195a71e7b46655 Mon Sep 17 00:00:00 2001 From: Maximellerbach Date: Fri, 27 Mar 2026 16:53:45 +0100 Subject: [PATCH] removing dataclass --- .../policies/groot/action_head/flow_matching_action_head.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/lerobot/policies/groot/action_head/flow_matching_action_head.py b/src/lerobot/policies/groot/action_head/flow_matching_action_head.py index 2c1ca6014..986820670 100644 --- a/src/lerobot/policies/groot/action_head/flow_matching_action_head.py +++ b/src/lerobot/policies/groot/action_head/flow_matching_action_head.py @@ -13,7 +13,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -from dataclasses import dataclass, field +from dataclasses import field from typing import TYPE_CHECKING import torch @@ -109,7 +109,6 @@ class MultiEmbodimentActionEncoder(nn.Module): return x -@dataclass class FlowmatchingActionHeadConfig(PretrainedConfig): """NOTE: N1.5 uses XEmbFlowmatchingPolicyHeadConfig as action head"""