feat(vis): increase animation FPS and rotation smoothness

Author: MichalRedm

src/visualization.py

@@ -213,12 +213,12 @@ def update(frame):
         ax.view_init(elev=20, azim=frame)
         return (fig,)
 
-    ani = FuncAnimation(fig, update, frames=np.arange(0, 360, 2), interval=50)
+    ani = FuncAnimation(fig, update, frames=np.arange(0, 360, 2), interval=40)
 
     if save_path:
         if save_path.endswith(".gif"):
             print("Generating 3D animation (GIF)... this may take a moment.")
-            ani.save(save_path, writer="pillow", fps=20)
+            ani.save(save_path, writer="pillow", fps=25)
         else:
             plt.savefig(save_path)
 
@@ -508,11 +508,11 @@ def update(frame):
         ax.view_init(elev=20, azim=frame)
         return (fig,)
 
-    ani = FuncAnimation(fig, update, frames=np.arange(0, 360, 4), interval=50)
+    ani = FuncAnimation(fig, update, frames=np.arange(0, 360, 2), interval=40)
 
     if save_path:
         if save_path.endswith(".gif"):
-            ani.save(save_path, writer="pillow", fps=15)
+            ani.save(save_path, writer="pillow", fps=25)
         else:
             plt.savefig(save_path)
 
@@ -787,7 +787,7 @@ def plot_two_variants_comparison_3d(
     fig = plt.figure(figsize=(20, 9))
     ax1 = fig.add_subplot(121, projection="3d")
     ax2 = fig.add_subplot(122, projection="3d")
-    
+
     # Define color palette
     c_ref = "#333333"
     c1 = "#1f77b4"
@@ -812,7 +812,7 @@ def format_label(name, m_dict):
             alpha=0.25,
             zorder=1,
         )
-    
+
     label1 = format_label(name1, metrics_dict1)
     ax1.scatter(
         metrics1["Return"],
@@ -847,7 +847,7 @@ def format_label(name, m_dict):
             alpha=0.25,
             zorder=1,
         )
-    
+
     label2 = format_label(name2, metrics_dict2)
     ax2.scatter(
         metrics2["Return"],
@@ -869,19 +869,23 @@ def format_label(name, m_dict):
     ax2.legend(loc="upper left", frameon=True, shadow=True, fontsize=10)
     ax2.view_init(elev=25, azim=45)
 
-    plt.suptitle("3D Pareto Front Quality Comparison (AWA Optimization)", fontsize=16, y=0.98)
+    plt.suptitle(
+        "3D Pareto Front Quality Comparison (AWA Optimization)", fontsize=16, y=0.98
+    )
     plt.tight_layout(rect=[0, 0.03, 1, 0.95])
 
     if save_path:
         if save_path.endswith(".gif"):
+
             def update(frame):
                 ax1.view_init(elev=25, azim=frame)
                 ax2.view_init(elev=25, azim=frame)
-                return fig,
+                return (fig,)
 
             from matplotlib.animation import FuncAnimation
-            ani = FuncAnimation(fig, update, frames=np.arange(0, 360, 5), interval=100)
-            ani.save(save_path, writer="pillow", dpi=100)
+
+            ani = FuncAnimation(fig, update, frames=np.arange(0, 360, 2), interval=40)
+            ani.save(save_path, writer="pillow", fps=25, dpi=100)
         else:
             plt.savefig(save_path, dpi=150, bbox_inches="tight")
     plt.close()