From 599b3a5d37f6fdd3f247a652e4d2578958e19e8c Mon Sep 17 00:00:00 2001 From: Purva Thakre Date: Mon, 1 Jul 2024 09:34:35 -0500 Subject: [PATCH] alessandro's feedback --- mitiq/lre/multivariate_scaling/layerwise_folding.py | 2 +- mitiq/lre/tests/test_layerwise_folding.py | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/mitiq/lre/multivariate_scaling/layerwise_folding.py b/mitiq/lre/multivariate_scaling/layerwise_folding.py index c3d7bfdf9b..d59cd8fd69 100644 --- a/mitiq/lre/multivariate_scaling/layerwise_folding.py +++ b/mitiq/lre/multivariate_scaling/layerwise_folding.py @@ -204,6 +204,6 @@ def multivariate_layer_scaling( folded_chunk_circ = folding_method(chunks_circ, scale_factor) folded_circuit += folded_chunk_circ _append_measurements(folded_circuit, terminal_measurements) - multiple_folded_circuits.append((folded_circuit)) + multiple_folded_circuits.append(folded_circuit) return multiple_folded_circuits diff --git a/mitiq/lre/tests/test_layerwise_folding.py b/mitiq/lre/tests/test_layerwise_folding.py index dc8c0ec461..8b1536ce74 100644 --- a/mitiq/lre/tests/test_layerwise_folding.py +++ b/mitiq/lre/tests/test_layerwise_folding.py @@ -83,9 +83,7 @@ def test_get_num_layers(test_input, expected): ) def test_get_num_chunks(test_input, test_chunks, expected_chunks): """Verifies the chunking function works as expected.""" - calculated_num_chunks = len(_get_chunks(test_input, test_chunks)) - - assert calculated_num_chunks == expected_chunks + assert expected_chunks == len(_get_chunks(test_input, test_chunks)) def test_layers_with_chunking():