Unverified Commit 7a976098 authored by Muhammed Fatih BALIN's avatar Muhammed Fatih BALIN Committed by GitHub
Browse files

[GraphBolt] Eliminate test warning (#7044)

parent f051a5cc
......@@ -72,6 +72,7 @@ class TorchBasedFeature(Feature):
def __init__(self, torch_feature: torch.Tensor, metadata: Dict = None):
super().__init__()
self._is_inplace_pinned = set()
assert isinstance(torch_feature, torch.Tensor), (
f"torch_feature in TorchBasedFeature must be torch.Tensor, "
f"but got {type(torch_feature)}."
......@@ -83,7 +84,6 @@ class TorchBasedFeature(Feature):
# Make sure the tensor is contiguous.
self._tensor = torch_feature.contiguous()
self._metadata = metadata
self._is_inplace_pinned = set()
def __del__(self):
# torch.Tensor.pin_memory() is not an inplace operation. To make it
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment