Skip to content

Commit

Permalink
black [skip ci]
Browse files Browse the repository at this point in the history
  • Loading branch information
dwierichs committed Jun 3, 2022
1 parent 774c317 commit 17044c9
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/transforms/test_batch_partial.py
Original file line number Diff line number Diff line change
Expand Up @@ -729,13 +729,14 @@ def test_different_batchdim_error():
"""Test that an error is raised when different batch
dimensions are given to the decorated QNode"""
dev = qml.device("default.qubit", wires=2)

# To test this error message, we need to use operations that do
# not report problematic broadcasting dimensions (in place of problematic
# batch dimensions) at tape creation. For this, we "delete" `ndim_params`.

class RX_no_ndim(qml.RX):
ndim_params = property(lambda self: self._ndim_params)

class RY_no_ndim(qml.RY):
ndim_params = property(lambda self: self._ndim_params)

Expand Down

0 comments on commit 17044c9

Please sign in to comment.