always remember to disable

PennyLaneAI · May 14, 2024 · faa5cf9 · faa5cf9
1 parent 39da90c
commit faa5cf9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pennylane/ops/functions/assert_valid.py b/pennylane/ops/functions/assert_valid.py
@@ -223,7 +223,8 @@ def _check_capture(op):
  " Please see the capture module documentation for more infromation."
  )
  ) from e
- qml.capture.disable()
+ finally:
+ qml.capture.disable()
 
 
 def _check_pickle(op):