Add error handling for unavailable plotting libraries in plot_wigner …

…function
LorenzoFioroni · Dec 9, 2024 · f2a6789 · f2a6789
1 parent 3429e7b
commit f2a6789
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/visualization.jl b/src/visualization.jl
@@ -25,3 +25,13 @@ plot_wigner(
     kwargs...,
 ) where {T,OpType<:Union{BraQuantumObject,KetQuantumObject,OperatorQuantumObject}} =
     plot_wigner(makeVal(library), state; kwargs...)
+
+plot_wigner(
+    ::Val{T},
+    state::QuantumObject{<:AbstractArray{T1},OpType};
+    kwargs...,
+) where {T,T1,OpType<:Union{BraQuantumObject,KetQuantumObject,OperatorQuantumObject}} = throw(
+    ArgumentError(
+        "The specified plotting library $T is not available. Try running `using $T` first.",
+    ),
+)