Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
[ghstack-poisoned]
  • Loading branch information
vmoens committed Nov 23, 2024
2 parents 2bc363f + 84caa00 commit bb97507
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions tensordict/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,6 @@ class _NoDefault(enum.IntEnum):


NO_DEFAULT = _NoDefault.ZERO
assert not NO_DEFAULT
T = TypeVar("T", bound="TensorDictBase")


Expand Down Expand Up @@ -4297,7 +4296,6 @@ def _view_and_pad(tensor):
elif k[-1].startswith("<NJT>"):
# NJT/NT always comes before offsets/shapes
nt = oldv
assert not v.numel()
nt_lengths = None
del flat_dict[k]
elif k[-1].startswith("<NJT_VALUES>"):
Expand Down Expand Up @@ -9453,7 +9451,6 @@ def _validate_value(
if device is not None and value.device != device:
if _device_recorder.marked and device.type != "cuda":
_device_recorder.record_transfer(device)
assert not non_blocking
value = value.to(device, non_blocking=non_blocking)
if check_shape:
if is_tc is None:
Expand Down

0 comments on commit bb97507

Please sign in to comment.