circuit: fix #240 trivial TN circuit rehearsal

jcmgray · Jul 3, 2024 · e20abf4 · e20abf4
1 parent cafc423
commit e20abf4
Showing 1 changed file with 1 addition and 1 deletion.
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)),
     }