Bugfix get_FIM

Hardcoded values have been changed.
Pyomo · Aug 20, 2024 · a68b280 · a68b280
1 parent 3af9d47
commit a68b280
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pyomo/contrib/doe/doe.py b/pyomo/contrib/doe/doe.py
@@ -53,6 +53,7 @@
 class ObjectiveLib(Enum):
     determinant = "determinant"
     trace = "trace"
+    minimum_eigenvalue = "minimum_eigenvalue"
     zero = "zero"
 
 
@@ -2043,8 +2044,8 @@ def get_FIM(self, model=None):
 
         # FIM is a lower triangular matrix for the optimal DoE problem.
         # Exploit symmetry to fill in the zeros.
-        for i in range(4):
-            for j in range(4):
+        for i in range(len(model.parameter_names)):
+            for j in range(len(model.parameter_names)):
                 if j < i:
                     fim_np[j, i] = fim_np[i, j]