diff --git a/pts/model/deepvar/deepvar_network.py b/pts/model/deepvar/deepvar_network.py index d37b0e5..1f403d5 100644 --- a/pts/model/deepvar/deepvar_network.py +++ b/pts/model/deepvar/deepvar_network.py @@ -422,7 +422,7 @@ class DeepVARTrainingNetwork(nn.Module): class DeepVARPredictionNetwork(DeepVARTrainingNetwork): - def __init__(self, num_parallel_samples: int, **kwargs) -> None: + def __init__(self, num_parallel_samples: int = 100, **kwargs) -> None: super().__init__(**kwargs) self.num_parallel_samples = num_parallel_samples