From 8a322da42292cd4b1c19a33821a8ee4c57b5a3d6 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Tue, 16 Apr 2024 16:35:04 +0100 Subject: [PATCH] backup wip --- tests/test_policies.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_policies.py b/tests/test_policies.py index 2547a3a..f53e402 100644 --- a/tests/test_policies.py +++ b/tests/test_policies.py @@ -4,7 +4,7 @@ import torch from lerobot.common.datasets.utils import cycle from lerobot.common.envs.utils import postprocess_action, preprocess_observation from lerobot.common.policies.factory import make_policy -from lerobot.common.policies.protocol import Policy +from lerobot.common.policies.policy_protocol import Policy from lerobot.common.envs.factory import make_env from lerobot.common.datasets.factory import make_dataset from lerobot.common.utils import init_hydra_config