From 967e2441516a1720737a9804e6392376a0bc30e0 Mon Sep 17 00:00:00 2001 From: r-sarma <126173968+r-sarma@users.noreply.github.com> Date: Mon, 14 Oct 2024 10:59:43 +0200 Subject: [PATCH] Update inference.py --- src/itwinai/torch/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/itwinai/torch/inference.py b/src/itwinai/torch/inference.py index c70474cb..ef10485a 100644 --- a/src/itwinai/torch/inference.py +++ b/src/itwinai/torch/inference.py @@ -265,7 +265,7 @@ def log( batch_idx=batch_idx, **kwargs ) - + @abc.abstractmethod def transform_predictions(self, batch: Batch) -> Batch: """