chore: move constants to utils (#2016)

This commit is contained in:
Steven Palma
2025-09-24 11:11:53 +02:00
committed by GitHub
parent c9787bd98a
commit 7cf04a5ec3
60 changed files with 74 additions and 74 deletions

View File

@@ -14,10 +14,6 @@
import pytest
import torch
from lerobot.constants import (
OPTIMIZER_PARAM_GROUPS,
OPTIMIZER_STATE,
)
from lerobot.optim.optimizers import (
AdamConfig,
AdamWConfig,
@@ -26,6 +22,10 @@ from lerobot.optim.optimizers import (
load_optimizer_state,
save_optimizer_state,
)
from lerobot.utils.constants import (
OPTIMIZER_PARAM_GROUPS,
OPTIMIZER_STATE,
)
@pytest.mark.parametrize(

View File

@@ -13,7 +13,6 @@
# limitations under the License.
from torch.optim.lr_scheduler import LambdaLR
from lerobot.constants import SCHEDULER_STATE
from lerobot.optim.schedulers import (
CosineDecayWithWarmupSchedulerConfig,
DiffuserSchedulerConfig,
@@ -21,6 +20,7 @@ from lerobot.optim.schedulers import (
load_scheduler_state,
save_scheduler_state,
)
from lerobot.utils.constants import SCHEDULER_STATE
def test_diffuser_scheduler(optimizer):