From 82a6b69e0e4a855d193be469a952f5b5413d87a1 Mon Sep 17 00:00:00 2001 From: AdilZouitine Date: Fri, 28 Mar 2025 13:26:47 +0000 Subject: [PATCH] Refactor imports in modeling_sac.py for improved organization - Rearranged import statements for better readability. - Removed unused imports and streamlined the code structure. --- lerobot/common/policies/sac/modeling_sac.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/lerobot/common/policies/sac/modeling_sac.py b/lerobot/common/policies/sac/modeling_sac.py index 3c9b26ec..e46cafcf 100644 --- a/lerobot/common/policies/sac/modeling_sac.py +++ b/lerobot/common/policies/sac/modeling_sac.py @@ -17,21 +17,19 @@ # TODO: (1) better device management -from dataclasses import asdict import math -from pathlib import Path -from typing import Callable, Dict, List, Optional, Tuple, Union +from dataclasses import asdict +from typing import Callable, List, Optional, Tuple import einops import numpy as np import torch import torch.nn as nn import torch.nn.functional as F # noqa: N812 -from huggingface_hub import PyTorchModelHubMixin from torch import Tensor -from lerobot.common.policies.pretrained import PreTrainedPolicy from lerobot.common.policies.normalize import Normalize, Unnormalize +from lerobot.common.policies.pretrained import PreTrainedPolicy from lerobot.common.policies.sac.configuration_sac import SACConfig from lerobot.common.policies.utils import get_device_from_parameters