allow retrieval of vector values on sweeps

convexengineering · Feb 17, 2022 · dd0ce90 · dd0ce90
1 parent 4301db7
commit dd0ce90
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gpkit/keydict.py b/gpkit/keydict.py
@@ -93,8 +93,8 @@ def __contains__(self, key):  # pylint:disable=too-many-return-statements
         if super().__contains__(key):  # pylint: disable=no-member
             if idx:
                 try:
-                    value = super().__getitem__(key)[idx]  # pylint: disable=no-member
-                    return True if is_sweepvar(value) else not isnan(value)
+                    val = super().__getitem__(key)[idx]  # pylint: disable=no-member
+                    return True if is_sweepvar(val) else not isnan(val).any()
                 except TypeError:
                     raise TypeError("%s has an idx, but its value in this"
                                     " KeyDict is the scalar %s."