Change SAC policy implementation with configuration and modeling classes
This commit is contained in:
committed by
Michel Aractingi
parent
875c0271b7
commit
760d60ad4b
@@ -64,7 +64,6 @@ def get_policy_class(name: str) -> PreTrainedPolicy:
|
||||
from lerobot.common.policies.sac.modeling_sac import SACPolicy
|
||||
|
||||
return SACPolicy, SACConfig
|
||||
|
||||
else:
|
||||
raise NotImplementedError(f"Policy with name {name} is not implemented.")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user