Skip to content

Commit

Permalink
move comment of tuple(t.shape) (#1287)
Browse files Browse the repository at this point in the history
  • Loading branch information
crcrpar authored Oct 11, 2024
1 parent a8762dd commit 16ad769
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions thunder/core/proxies.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import copy
from enum import auto, Enum
from numbers import Number
from typing import Type, Optional, Any, Tuple, List, Union
from typing import Any
from collections.abc import Callable
from collections.abc import Sequence

Expand Down Expand Up @@ -1974,8 +1974,8 @@ def tensorproxy(t: torch.Tensor, /, *, name: None | str, history: None | tuple =
for idx, s in enumerate(t.shape)
)
else:
# NOTE Without tuple(t.shape) then the shape would be a torch.Size object
shape = tuple(t.shape)
# NOTE Without tuple(t.shape) then the shape would be a torch.Size object
return TensorProxy(
name,
shape=tuple(shape),
Expand Down

0 comments on commit 16ad769

Please sign in to comment.