Fix changed name

dbhart · Nov 17, 2023 · e07e292 · e07e292
1 parent d347ad6
commit e07e292
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/wntr/network/model.py b/wntr/network/model.py
@@ -336,9 +336,9 @@ def msx(self):
     def msx(self, msx):
         if msx is None:
             self._msx = None
-        from wntr.msx.base import AbstractQualityModel
-        if not isinstance(msx, AbstractQualityModel):
-            raise TypeError('Expected AbstractQualityModel (or derived), got {}'.format(type(msx)))
+        from wntr.msx.base import QualityModelBase
+        if not isinstance(msx, QualityModelBase):
+            raise TypeError('Expected QualityModelBase (or derived), got {}'.format(type(msx)))
 
     def add_msx_model(self, msx_filename=None):
         """Add an msx model from a MSX input file (.msx extension)"""