Fix viz.draw (python-graphblas#320)

jim22k · Nov 8, 2022 · eaf0cbb · eaf0cbb
1 parent ec61553
commit eaf0cbb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/graphblas/viz.py b/graphblas/viz.py
@@ -58,7 +58,8 @@ def draw(m):  # pragma: no cover
     except ImportError as exc:
         print(exc.msg)
         return
-    if not isinstance(_output_type(m), (_Matrix, _TransposedMatrix)):
+    typ = _output_type(m)
+    if typ is not _Matrix and typ is not _TransposedMatrix:
         print(f"Can only draw a Matrix, not {type(m)}")
         return