Handle new config with sac

2026-05-27 14:39:43 +00:00 · 2025-03-24 20:19:28 +00:00
parent b2025b852c
commit f483931fc0
4 changed files with 242 additions and 240 deletions
@@ -55,10 +55,9 @@ def get_policy_class(name: str) -> PreTrainedPolicy:

        return PI0Policy
    elif name == "sac":
-        from lerobot.common.policies.sac.configuration_sac import SACConfig
        from lerobot.common.policies.sac.modeling_sac import SACPolicy

-        return SACPolicy, SACConfig
+        return SACPolicy
    else:
        raise NotImplementedError(f"Policy with name {name} is not implemented.")