fix ham terms drawing

jcmgray · Jul 10, 2024 · ae6266b · ae6266b
1 parent 43330c9
commit ae6266b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/quimb/tensor/drawing.py b/quimb/tensor/drawing.py
@@ -1331,7 +1331,7 @@ def get_positions(
     iterations="auto",
     k=None,
 ):
-    if (tn.num_tensors == 1) and (fix is None):
+    if (tn is not None) and (tn.num_tensors == 1) and (fix is None):
         # single tensor, layout manually
         return layout_single_tensor(tn, dim=dim)