🚧

understandable-machine-intelligence-lab · Apr 28, 2024 · 4157a5e · 4157a5e
1 parent 1b7bb85
commit 4157a5e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/quantus/helpers/model/pytorch_model.py b/quantus/helpers/model/pytorch_model.py
@@ -121,7 +121,7 @@ def _obtain_predictions(
             npt.ArrayLike,
             Mapping[str, Union[torch.Tensor, npt.ArrayLike]],
         ],
-        model_predict_kwargs: dict[str, Any],
+        model_predict_kwargs: Dict[str, Any],
     ) -> torch.Tensor:
         if safe_isinstance(self.model, "transformers.modeling_utils.PreTrainedModel"):