Replace domain comparison by type(domain) comparison

Fixes Pyomo#995
RomeoV · May 2, 2019 · 4b97240 · 4b97240
1 parent 41ef9ce
commit 4b97240
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyomo/core/plugins/transform/discrete_vars.py b/pyomo/core/plugins/transform/discrete_vars.py
@@ -114,7 +114,7 @@ def _apply_to(self, model, **kwds):
         _base_model_vars = model.component_data_objects(
             Var, active=True, descend_into=True )
         for var in _base_model_vars:
-            if var.domain in _discrete_relaxation_map and not var.is_fixed():
+            if any(type(var.domain) is type(domain) for domain in _discrete_relaxation_map) and not var.is_fixed():
                 fixed_vars.append(var)
                 var.fix()
         model._fixed_discrete_vars = Suffix(direction=Suffix.LOCAL)