From 2524c1e0a4286eaa25bd1832216a30eea6aaeae1 Mon Sep 17 00:00:00 2001
From: Amogh Joshi <joshi.amoghn@gmail.com>
Date: Wed, 1 May 2024 15:50:09 -0400
Subject: [PATCH] Fix sample visualization, remove existing stats for
 rangeland_weed_australia

---
 agml/_assets/public_datasources.json | 12 ------------
 agml/viz/labels.py                   |  4 ++++
 2 files changed, 4 insertions(+), 12 deletions(-)

diff --git a/agml/_assets/public_datasources.json b/agml/_assets/public_datasources.json
index 806e6b7b..5903a8c3 100644
--- a/agml/_assets/public_datasources.json
+++ b/agml/_assets/public_datasources.json
@@ -254,18 +254,6 @@
         "annotation_format": "directory_names",
         "n_images": "17509",
         "docs_url": "https://github.com/AlexOlsen/DeepWeeds",
-        "stats": {
-            "mean": [
-                0.3785816431045532,
-                0.38957422971725464,
-                0.3797682821750641
-            ],
-            "std": [
-                0.22421954572200775,
-                0.22450360655784607,
-                0.22274591028690338
-            ]
-        },
         "classes": {
             "0": "Chinee Apple",
             "1": "Lantana",
diff --git a/agml/viz/labels.py b/agml/viz/labels.py
index ce6629ea..029fceac 100644
--- a/agml/viz/labels.py
+++ b/agml/viz/labels.py
@@ -116,7 +116,11 @@ def show_images_and_labels(images,
         plt.setp(ax.spines.values(), visible = False)
         ax.set_xlabel(label)
 
+        # decrease the label size (if it's too big)
+        ax.xaxis.label.set_size(8)
+
     # Display and return the image.
+    fig.tight_layout()
     image = convert_figure_to_image()
     if not kwargs.get('no_show', False):
         _ = display_image(image)