summaryrefslogtreecommitdiff
path: root/python/variable.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/variable.py')
-rw-r--r--python/variable.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/python/variable.py b/python/variable.py
index 564281c6..07b8863c 100644
--- a/python/variable.py
+++ b/python/variable.py
@@ -841,7 +841,7 @@ class ParameterVariables:
self._func = func
def __repr__(self):
- return repr(self._vars)
+ return f"<ParameterVariables: {str(self._vars)}>"
def __len__(self):
return len(self._vars)
@@ -850,6 +850,9 @@ class ParameterVariables:
for var in self._vars:
yield var
+ def __eq__(self, other) -> bool:
+ return (self._vars, self._confidence, self._func) == (other._vars, other._confidence, other._func)
+
def __getitem__(self, idx) -> 'Variable':
return self._vars[idx]