From c5f962b9fa5eae29967c9ff05fc457c8141e935d Mon Sep 17 00:00:00 2001 From: obliviateandsurrender Date: Fri, 13 Sep 2024 23:40:07 -0400 Subject: [PATCH] fix `isort` --- tests/devices/qubit/test_simulate.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/devices/qubit/test_simulate.py b/tests/devices/qubit/test_simulate.py index 18483550cd4..256ea48125d 100644 --- a/tests/devices/qubit/test_simulate.py +++ b/tests/devices/qubit/test_simulate.py @@ -22,15 +22,15 @@ import pennylane as qml from pennylane.devices.qubit import get_final_state, measure_final_state, simulate from pennylane.devices.qubit.simulate import ( + TreeTraversalStack, _FlexShots, branch_state, combine_measurements_core, + counts_to_probs, find_post_processed_mcms, samples_to_counts, - counts_to_probs, simulate_tree_mcm, split_circuit_at_mcms, - TreeTraversalStack, ) @@ -1205,10 +1205,10 @@ def test_init_with_depth(self, max_depth): tree_stack = TreeTraversalStack(max_depth) assert tree_stack.counts.count(None) == max_depth - assert tree_stack.counts.probs(None) == max_depth - assert tree_stack.counts.results_0(None) == max_depth - assert tree_stack.counts.results_1(None) == max_depth - assert tree_stack.counts.states(None) == max_depth + assert tree_stack.probs.count(None) == max_depth + assert tree_stack.results_0.count(None) == max_depth + assert tree_stack.results_1.count(None) == max_depth + assert tree_stack.states.count(None) == max_depth @pytest.mark.unit def test_full_prune_empty_methods(self):