From 7d868725fab8a768bc6c8c87923b144ea2e194ac Mon Sep 17 00:00:00 2001 From: Matthias Hadlich Date: Thu, 5 Oct 2023 15:52:19 +0200 Subject: [PATCH] Add track_meta=True to Signalfillempty Signed-off-by: Matthias Hadlich --- monai/transforms/signal/array.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/monai/transforms/signal/array.py b/monai/transforms/signal/array.py index 1a7b87a774..938f42192c 100644 --- a/monai/transforms/signal/array.py +++ b/monai/transforms/signal/array.py @@ -380,7 +380,7 @@ def __call__(self, signal: NdarrayOrTensor) -> NdarrayOrTensor: Args: signal: signal to be filled """ - signal = torch.nan_to_num(convert_to_tensor(signal), nan=self.replacement) + signal = torch.nan_to_num(convert_to_tensor(signal, track_meta=True), nan=self.replacement) return signal