hiro/hopsflow
1
0
Fork 0
mirror of https://github.com/vale981/hopsflow synced 2025-03-04 16:31:38 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

do not allow adding values with different bath numbers

Browse source
This commit is contained in:
Valentin Boettcher 2022-11-29 18:11:07 -05:00
parent e2f1c76090
commit f8b554fcfe
No known key found for this signature in database
GPG key ID: E034E12B7AF56ACE
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
hopsflow/util.py
View file

@ -189,10 +189,15 @@ class EnsembleValue:
def __add__(
self, other: Union["EnsembleValue", float, int, np.ndarray]
) -> EnsembleValue:
if type(self) == type(other):
if isinstance(other, EnsembleValue):
if len(self) != len(other):
raise RuntimeError("Can only add values of equal length.")
if self.num_baths != other.num_baths:
raise RuntimeError(
"Can only add values pertaining to an equal number of baths."
)
left = self._value
right = other._value