NLv2: only locate / process Suffixes in the context of the model bein…

…g written
Pyomo · Aug 13, 2024 · bc18546 · bc18546
1 parent 89994ae
commit bc18546
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyomo/repn/plugins/nl_writer.py b/pyomo/repn/plugins/nl_writer.py
@@ -510,8 +510,8 @@ def compile(self, column_order, row_order, obj_order, model_id):
 class CachingNumericSuffixFinder(SuffixFinder):
     scale = True
 
-    def __init__(self, name, default=None):
-        super().__init__(name, default)
+    def __init__(self, name, default=None, context=None):
+        super().__init__(name, default, context)
         self.suffix_cache = {}
 
     def __call__(self, obj):
@@ -646,7 +646,7 @@ def write(self, model):
         # Data structures to support variable/constraint scaling
         #
         if self.config.scale_model and 'scaling_factor' in suffix_data:
-            scaling_factor = CachingNumericSuffixFinder('scaling_factor', 1)
+            scaling_factor = CachingNumericSuffixFinder('scaling_factor', 1, model)
             scaling_cache = scaling_factor.suffix_cache
             del suffix_data['scaling_factor']
         else: