Fix unused import warnings for ConcentrationPolarizationType and Mass…

…TransferCoefficient
watertap-org · Apr 27, 2023 · 212059f · 212059f
1 parent d143c6f
commit 212059f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/watertap/unit_models/reverse_osmosis_0D.py b/watertap/unit_models/reverse_osmosis_0D.py
@@ -32,9 +32,12 @@
     _add_has_full_reporting,
 )
 
-
 __author__ = "Tim Bartholomew, Adam Atia"
 
+# Allow users to import ConcentrationPolarizationType from here instead of
+# from watertap.core, this also avoids an unused import warning from linters
+ConcentrationPolarizationType = ConcentrationPolarizationType
+
 
 @declare_process_block_class("ReverseOsmosis0D")
 class ReverseOsmosisData(ReverseOsmosisBaseData):
@@ -72,7 +75,6 @@ def _add_deltaP(self):
         add_object_reference(self, "deltaP", self.feed_side.deltaP)
 
     def _add_mass_transfer(self):
-
         units_meta = self.config.property_package.get_metadata().get_derived_units
 
         # not in 1DRO

diff --git a/watertap/unit_models/reverse_osmosis_1D.py b/watertap/unit_models/reverse_osmosis_1D.py
@@ -40,6 +40,12 @@
 
 __author__ = "Adam Atia"
 
+# Allow users to import the ConcentrationPolarizationType enum and the
+# MassTransferCoefficient enm from here instead of from watertap.core, this
+# also avoids an unused import warning from linters
+ConcentrationPolarizationType = ConcentrationPolarizationType
+MassTransferCoefficient = MassTransferCoefficient
+
 # Set up logger
 _log = idaeslog.getLogger(__name__)