all tests run, now autoset lineage

convexengineering · Jan 7, 2022 · 406c56c · 406c56c
1 parent 48474c5
commit 406c56c
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/gpkit/solution_array.py b/gpkit/solution_array.py
@@ -647,8 +647,8 @@ def savetxt(self, filename="solution.txt", *, printmodel=True, **kwargs):
     def savejson(self, filename="solution.json", showvars=None):
         "Saves solution table as a json file"
         sol_dict = {}
-        for key in self.name_collision_varkeys():
-            key.descr["necessarylineage"] = True
+        if self._lineageset:
+            self.set_necessarylineage(clear=True)
         data = self["variables"]
         if showvars:
             showvars = self._parse_showvars(showvars)
@@ -661,8 +661,6 @@ def savejson(self, filename="solution.json", showvars=None):
             else:
                 val = {"v": v, "u": k.unitstr()}
             sol_dict[key] = val
-        for key in self.name_collision_varkeys():
-            del key.descr["necessarylineage"]
         with open(filename, "w") as f:
             json.dump(sol_dict, f)