wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Refactor imports in modeling_sac.py for improved organization 

- Rearranged import statements for better readability.
- Removed unused imports and streamlined the code structure.
This commit is contained in:
AdilZouitine 2025-03-28 13:26:47 +00:00
parent 6f7024242a
commit 82a6b69e0e
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lerobot/common/policies/sac/modeling_sac.py
View File

@ -17,21 +17,19 @@
# TODO: (1) better device management # TODO: (1) better device management
from dataclasses import asdict
import math import math
from pathlib import Path from dataclasses import asdict
from typing import Callable, Dict, List, Optional, Tuple, Union from typing import Callable, List, Optional, Tuple
import einops import einops
import numpy as np import numpy as np
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F # noqa: N812 import torch.nn.functional as F # noqa: N812
from huggingface_hub import PyTorchModelHubMixin
from torch import Tensor from torch import Tensor
from lerobot.common.policies.pretrained import PreTrainedPolicy
from lerobot.common.policies.normalize import Normalize, Unnormalize 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.sac.configuration_sac import SACConfig
from lerobot.common.policies.utils import get_device_from_parameters from lerobot.common.policies.utils import get_device_from_parameters