From c10ac2b5605f1b0034f3b456150e68c78858b0c6 Mon Sep 17 00:00:00 2001
From: Lorenzo Fioroni <lorenzo.fioroni@epfl.ch>
Date: Wed, 11 Dec 2024 09:03:25 +0100
Subject: [PATCH] Update CairoMakie imports and change axis labels

---
 ext/QuantumToolboxCairoMakieExt.jl | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/ext/QuantumToolboxCairoMakieExt.jl b/ext/QuantumToolboxCairoMakieExt.jl
index 9e86e761..16de321c 100644
--- a/ext/QuantumToolboxCairoMakieExt.jl
+++ b/ext/QuantumToolboxCairoMakieExt.jl
@@ -1,7 +1,7 @@
 module QuantumToolboxCairoMakieExt
 
 using QuantumToolbox
-using CairoMakie
+using CairoMakie: Axis, Axis3, Colorbar, Figure, GridLayout, heatmap!, surface!, GridPosition, @L_str, Reverse
 
 @doc raw"""
     plot_wigner(
@@ -100,8 +100,8 @@ function _plot_wigner(
         Colorbar(lyt[1, 2], hm)
     end
 
-    ax.xlabel = L"\Re(\alpha)"
-    ax.ylabel = L"\Im(\alpha)"
+    ax.xlabel = L"\textrm{Re}(\alpha)"
+    ax.ylabel = L"\textrm{Im}(\alpha)"
     return fig, ax, hm
 end
 
@@ -133,8 +133,8 @@ function _plot_wigner(
         Colorbar(lyt[1, 2], surf)
     end
 
-    ax.xlabel = L"\Re(\alpha)"
-    ax.ylabel = L"\Im(\alpha)"
+    ax.xlabel = L"\textrm{Re}(\alpha)"
+    ax.ylabel = L"\textrm{Im}(\alpha)"
     ax.zlabel = "Wigner function"
     return fig, ax, surf
 end