Cosmetic changes to the tile filtering example

NVIDIA · Dec 2, 2024 · 2c3d619 · 2c3d619
1 parent dc488eb
commit 2c3d619
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/warp/examples/tile/example_tile_filtering.py b/warp/examples/tile/example_tile_filtering.py
@@ -85,14 +85,21 @@ def conv_tiled(x: wp.array2d(dtype=wp.vec2d), y: wp.array2d(dtype=wp.vec2d), z:
 try:
     import matplotlib.pyplot as plt
 
-    plt.figure()
-    plt.plot(
+    fig, ax = plt.subplots(figsize=(10, 5))
+
+    ax.plot(
         x,
-        "r-",
+        color="#DDDDDD",
+        linewidth=2,
         label="Original",
     )
-    plt.plot(y_test[0, :].real, "b.-", label="Smoothed")
-    plt.legend()
-    plt.savefig("filter.png")
+    ax.plot(y_test[0, :].real, color="#76B900", linewidth=3, label="Smoothed")
+
+    ax.legend()
+    ax.grid(True)
+
+    plt.tight_layout()
+    plt.show()
+
 except ModuleNotFoundError:
     print("Matplotlib not available; skipping figure")