Merge pull request #22 from Nomos11/plotting_fix

Plotting fix
qutech · Apr 25, 2023 · a3b2ae1 · a3b2ae1
2 parents 218895e + b32883a
commit a3b2ae1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/qopt/plotting.py b/qopt/plotting.py
@@ -39,7 +39,6 @@
 
 import numpy as np
 import matplotlib.pyplot as plt
-from mpl_toolkits import mplot3d
 from unittest import mock
 from warnings import warn
 from typing import Sequence
@@ -99,7 +98,7 @@ def plot_bloch_vector_evolution(
     figsize = bloch_kwargs.pop('figsize', [5, 5])
     view = bloch_kwargs.pop('view', [-60, 30])
     fig = plt.figure(figsize=figsize)
-    axes = mplot3d.Axes3D(fig, azim=view[0], elev=view[1])
+    axes = fig.add_subplot(projection='3d', azim=view[0], elev=view[1])
     bloch_kwargs.setdefault('view', [-150, 30])
     b = qt.Bloch(fig=fig, axes=axes, **bloch_kwargs)