From 59397fb44afc19297fcdb57dbed2ddef7a6e34ca Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Sat, 9 Mar 2024 18:44:36 +0100 Subject: [PATCH] Move tdmpc files --- lerobot/common/policies/factory.py | 2 +- lerobot/common/policies/tdmpc/__init__.py | 0 lerobot/common/policies/{tdmpc_helper.py => tdmpc/helper.py} | 0 lerobot/common/policies/{tdmpc.py => tdmpc/policy.py} | 2 +- 4 files changed, 2 insertions(+), 2 deletions(-) create mode 100644 lerobot/common/policies/tdmpc/__init__.py rename lerobot/common/policies/{tdmpc_helper.py => tdmpc/helper.py} (100%) rename lerobot/common/policies/{tdmpc.py => tdmpc/policy.py} (99%) diff --git a/lerobot/common/policies/factory.py b/lerobot/common/policies/factory.py index 9507586c..a956cb4b 100644 --- a/lerobot/common/policies/factory.py +++ b/lerobot/common/policies/factory.py @@ -1,6 +1,6 @@ def make_policy(cfg): if cfg.policy.name == "tdmpc": - from lerobot.common.policies.tdmpc import TDMPC + from lerobot.common.policies.tdmpc.policy import TDMPC policy = TDMPC(cfg.policy, cfg.device) elif cfg.policy.name == "diffusion": diff --git a/lerobot/common/policies/tdmpc/__init__.py b/lerobot/common/policies/tdmpc/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/lerobot/common/policies/tdmpc_helper.py b/lerobot/common/policies/tdmpc/helper.py similarity index 100% rename from lerobot/common/policies/tdmpc_helper.py rename to lerobot/common/policies/tdmpc/helper.py diff --git a/lerobot/common/policies/tdmpc.py b/lerobot/common/policies/tdmpc/policy.py similarity index 99% rename from lerobot/common/policies/tdmpc.py rename to lerobot/common/policies/tdmpc/policy.py index 42fbb825..ae9888a5 100644 --- a/lerobot/common/policies/tdmpc.py +++ b/lerobot/common/policies/tdmpc/policy.py @@ -8,7 +8,7 @@ import numpy as np import torch import torch.nn as nn -import lerobot.common.policies.tdmpc_helper as h +import lerobot.common.policies.tdmpc.helper as h FIRST_FRAME = 0