Skip to content

Commit d68f7c3

Browse files
committed
Use defaultdict for desymmetrized dictionary creation.
1 parent eaa5352 commit d68f7c3

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

inflation/lp/InflationLP.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -898,7 +898,7 @@ def desymmetrize_certificate(self) -> dict:
898898
raise Exception("For extracting a certificate you need to solve " +
899899
"a problem. Call \"InflationSDP.solve()\" first.")
900900

901-
desymmetrized = {}
901+
desymmetrized = defaultdict(int)
902902
norm = len(self.InflationProblem.symmetries)
903903
lexmon_names = self.InflationProblem._lexrepr_to_copy_index_free_names
904904
for symm in self.InflationProblem.symmetries:
@@ -911,11 +911,8 @@ def desymmetrize_certificate(self) -> dict:
911911
desymm_name = "P[" + " ".join(desymm_mon) + "]"
912912
else:
913913
desymm_name = "1"
914-
if desymm_name not in desymmetrized:
915-
desymmetrized[desymm_name] = coeff / norm
916-
else:
917-
desymmetrized[desymm_name] += coeff / norm
918-
return desymmetrized
914+
desymmetrized[desymm_name] += coeff / norm
915+
return {k:v for k,v in desymmetrized.items()}
919916

920917
###########################################################################
921918
# OTHER ROUTINES EXPOSED TO THE USER #

0 commit comments

Comments
 (0)