double decorate

Lightning-AI · Nov 13, 2020 · f3b9c38 · f3b9c38
1 parent 3ef3dab
commit f3b9c38
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/models/test_hparams.py b/tests/models/test_hparams.py
@@ -55,6 +55,7 @@ def wrapper(*args, **kwargs):
 class SaveHparamsDecoratedModel(BoringModel):
  """ Tests that a model can take an object """
  @decorate
+ @decorate
  def __init__(self, hparams, *my_args, **my_kwargs):
  super().__init__()
  self.save_hyperparameters(hparams)
@@ -63,6 +64,7 @@ def __init__(self, hparams, *my_args, **my_kwargs):
 class AssignHparamsDecoratedModel(BoringModel):
  """ Tests that a model can take an object with explicit setter"""
  @decorate
+ @decorate
  def __init__(self, hparams, *my_args, **my_kwargs):
  super().__init__()
  self.hparams = hparams