From 02cbcb83074d668008d25c46b2163adad482c327 Mon Sep 17 00:00:00 2001 From: davidwilby <24752124+davidwilby@users.noreply.github.com> Date: Thu, 13 Jun 2024 16:38:40 +0100 Subject: [PATCH] use python 3.8 compatible typing --- deepsensor/model/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deepsensor/model/model.py b/deepsensor/model/model.py index d32d0f07..cb758462 100644 --- a/deepsensor/model/model.py +++ b/deepsensor/model/model.py @@ -243,7 +243,7 @@ def predict( aux_at_targets_override: Union[xr.Dataset, xr.DataArray] = None, aux_at_targets_override_is_normalised: bool = False, resolution_factor: int = 1, - pred_params: tuple[str] = ("mean", "std"), + pred_params: Tuple[str] = ("mean", "std"), n_samples: int = 0, ar_sample: bool = False, ar_subsample_factor: int = 1, @@ -274,7 +274,7 @@ def predict( Whether the `aux_at_targets_override` coords are normalised. If False, the DataProcessor will normalise the coords before passing to model. Default False. - pred_params (tuple[str]): + pred_params (Tuple[str]): Tuple of prediction parameters to return. The strings refer to methods of the model class which will be called and stored in the Prediction object. Default ("mean", "std").