Skip to content

Commit

Permalink
[Refactor] tensordict._tensordict -> tensordict._C (#2286)
Browse files Browse the repository at this point in the history
  • Loading branch information
vmoens authored Jul 10, 2024
1 parent a0a47a9 commit 28acf61
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion test/test_cost.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
from dataclasses import asdict, dataclass

from packaging import version as pack_version
from tensordict._tensordict import unravel_keys
from tensordict._C import unravel_keys

from tensordict.nn import (
InteractionType,
Expand Down
2 changes: 1 addition & 1 deletion torchrl/envs/transforms/transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@
unravel_key,
unravel_key_list,
)
from tensordict._tensordict import _unravel_key_to_tuple
from tensordict._C import _unravel_key_to_tuple
from tensordict.nn import dispatch, TensorDictModuleBase
from tensordict.utils import expand_as_right, expand_right, NestedKey
from torch import nn, Tensor
Expand Down

0 comments on commit 28acf61

Please sign in to comment.