From d3934334f956f5803001b066688a072e290d63fb Mon Sep 17 00:00:00 2001 From: Kevin Hsieh Date: Thu, 26 Dec 2024 17:57:52 -0800 Subject: [PATCH] Fix handling for to_custom_tensor Signed-off-by: Kevin Hsieh --- .../torch/src/python/aimet_torch/custom/custom_tensor_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TrainingExtensions/torch/src/python/aimet_torch/custom/custom_tensor_utils.py b/TrainingExtensions/torch/src/python/aimet_torch/custom/custom_tensor_utils.py index 29d13d4e038..2682d0329d7 100644 --- a/TrainingExtensions/torch/src/python/aimet_torch/custom/custom_tensor_utils.py +++ b/TrainingExtensions/torch/src/python/aimet_torch/custom/custom_tensor_utils.py @@ -45,7 +45,7 @@ def to_torch_tensor(tensors): def to_custom_tensor(original, torch_tensors): """ placeholder in case spconv doesn't exist """ - return original + return torch_tensors else: from typing import List, Union, Tuple import torch