fix: probabilities results in case of prob_type=U

This commit is contained in:
MatteoRobbiati
2025-02-03 17:18:48 +01:00
parent 08ad49eb19
commit 1036075c9c

View File

@@ -23,7 +23,16 @@ class TensorNetworkResult:
def probabilities(self):
"""Return calculated probabilities according to the given method."""
return self.measured_probabilities[self.prob_type]
if self.prob_type == "U":
for bitstring in self.measured_probabilities[self.prob_type]:
self.measured_probabilities[self.prob_type][bitstring] = (
self.measured_probabilities[self.prob_type][bitstring][1]
- self.measured_probabilities[self.prob_type][bitstring][0]
)
probabilities = self.measured_probabilities[self.prob_type]
else:
probabilities = self.measured_probabilities[self.prob_type]
return probabilities
def frequencies(self):
"""Return frequencies if a certain number of shots has been set."""