style fixes

Qiskit · Oct 4, 2019 · c9d5db5 · c9d5db5
1 parent 6db1efc
commit c9d5db5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/qiskit/providers/aer/extensions/snapshot_probabilities.py b/qiskit/providers/aer/extensions/snapshot_probabilities.py
@@ -32,11 +32,7 @@ def __init__(self, label, num_qubits, variance=False):
         Raises:
             ExtensionError: if snapshot is invalid.
         """
-        if variance:
-            snapshot_type = 'probabilities_with_variance'
-        else:
-            snapshot_type = 'probabilities'
-
+        snapshot_type = 'probabilities_with_variance' if variance else 'probabilities'
         super().__init__(label, snapshot_type=snapshot_type,
                          num_qubits=num_qubits)
 

diff --git a/test/terra/backends/qasm_simulator/qasm_snapshot.py b/test/terra/backends/qasm_simulator/qasm_snapshot.py
@@ -504,11 +504,8 @@ def probabilitiy_snapshots(data, labels):
         output = {}
         for label in labels:
             snaps = data.get("snapshots", {}).get("probabilities", {}).get(label, [])
-            # Convert list into dict
-            inner = {}
-            for snap_dict in snaps:
-                inner[snap_dict['memory']] = snap_dict['value']
-            output[label] = inner
+            output[label] = {snap_dict['memory']: snap_dict['value']
+                             for snap_dict in snaps}
         return output
 
     def test_snapshot_probabilities_pre_measure(self):