circuit: fix #240 trivial TN circuit rehearsal

jcmgray · jcmgray · commit e20abf46c813 · 2024-07-03T15:10:04.000-07:00
diff --git a/quimb/tensor/circuit.py b/quimb/tensor/circuit.py
@@ -1350,7 +1350,7 @@ def rehearsal_dict(tn, tree):
         "tn": tn,
         "tree": tree,
         "W": tree.contraction_width(),
-        "C": math.log10(tree.contraction_cost()),
+        "C": math.log10(max(tree.contraction_cost(), 1)),
     }
 
 

Original file line number	Diff line number	Diff line change
`@@ -1350,7 +1350,7 @@ def rehearsal_dict(tn, tree):`
`1350`	`1350`	`"tn": tn,`
`1351`	`1351`	`"tree": tree,`
`1352`	`1352`	`"W": tree.contraction_width(),`
`1353`		`- "C": math.log10(tree.contraction_cost()),`
	`1353`	`+ "C": math.log10(max(tree.contraction_cost(), 1)),`
`1354`	`1354`	`}`
`1355`	`1355`
`1356`	`1356`