add kwargs to post_process_samples

awslabs · Jul 12, 2023 · 59a49fe · 59a49fe
1 parent 6524fec
commit 59a49fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gluonts/mx/model/deepvar/_network.py b/src/gluonts/mx/model/deepvar/_network.py
@@ -744,7 +744,7 @@ def predict_hybrid_forward(
  begin_states=state,
  )
 
- def post_process_samples(self, samples: Tensor) -> Tensor:
+ def post_process_samples(self, samples: Tensor, **kwargs) -> Tensor:
  """
  Method to enforce domain-specific constraints on the generated samples.