diff --git a/src/View/Ensembles/Edit/Window.py b/src/View/Ensembles/Edit/Window.py index 077f001a..ab9da3ec 100644 --- a/src/View/Ensembles/Edit/Window.py +++ b/src/View/Ensembles/Edit/Window.py @@ -52,6 +52,7 @@ class StricklersCoefModelItem(PamhyrModelItem): def get_value(self): return (self._data, self.parent()._data) + class StricklersModelItem(PamhyrModelItem): def __init__(self, data=None, trad=None, parent=None): super(StricklersModelItem, self).__init__( @@ -67,6 +68,7 @@ class StricklersModelItem(PamhyrModelItem): def get_value(self): return self._data + class EnsembleEditDialog(PamhyrDialog): _pamhyr_ui = "EnsembleEditDialog" _pamhyr_name = "Parameters" @@ -137,7 +139,7 @@ class EnsembleEditDialog(PamhyrDialog): self._tree.itemClicked.connect(self._tree_item_select) def _tree_item_select(self, item): - if type(item) == StricklersCoefModelItem: + if type(item) is StricklersCoefModelItem: self._target_data = item.get_value() def update_distribution(self, txt):