From a387f1ee21bc6f748c617b386573889515861b11 Mon Sep 17 00:00:00 2001
From: SongshGeo <songshgeo@gmail.com>
Date: Sat, 17 Jan 2026 17:43:50 +0100
Subject: [PATCH] fix(tracker): :bug: Rename 'distribution' to 'samples' in
 AimTracker for API compatibility

---
 abses/utils/tracker/aim_tracker.py | 6 +++---
 tests/utils/test_aim_tracker.py    | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/abses/utils/tracker/aim_tracker.py b/abses/utils/tracker/aim_tracker.py
index 9bece65e..5e2a9283 100644
--- a/abses/utils/tracker/aim_tracker.py
+++ b/abses/utils/tracker/aim_tracker.py
@@ -172,7 +172,7 @@ def log_agent_vars(
                     self._run.track(numeric_series.iloc[0], name=metric_name, step=step)
                 else:
                     dist = Distribution(
-                        distribution=numeric_series, bin_count=self._bin_count
+                        samples=numeric_series, bin_count=self._bin_count
                     )
                     self._run.track(dist, name=metric_name, step=step)
                 # Additional statistics
@@ -196,7 +196,7 @@ def log_agent_vars(
                 else:
                     # Multiple values, use Distribution
                     dist = Distribution(
-                        distribution=numeric_series, bin_count=self._bin_count
+                        samples=numeric_series, bin_count=self._bin_count
                     )
                     self._run.track(dist, name=metric_name, step=step)
 
@@ -258,7 +258,7 @@ def log_agent_vars(
                             )
                         else:
                             dist = Distribution(
-                                distribution=numeric_series, bin_count=self._bin_count
+                                samples=numeric_series, bin_count=self._bin_count
                             )
                             self._run.track(dist, name=metric_name, step=step)
                 except (TypeError, ValueError):
diff --git a/tests/utils/test_aim_tracker.py b/tests/utils/test_aim_tracker.py
index 34f3e2a9..63d9dda2 100644
--- a/tests/utils/test_aim_tracker.py
+++ b/tests/utils/test_aim_tracker.py
@@ -42,7 +42,7 @@ def mock_distribution_class():
 
     # Create a mock Distribution class when aim is not available
     # It should be callable (a class), not an instance
-    def create_mock_distribution(distribution, bin_count=64):
+    def create_mock_distribution(samples, bin_count=64):
         """Create a mock Distribution instance."""
         mock_instance = MagicMock()
         mock_instance.bin_count = bin_count
@@ -92,7 +92,7 @@ def aim_tracker_custom_bin(mock_aim_run):
         dist_class = Distribution
     else:
 
-        def create_mock_distribution(distribution, bin_count=100):
+        def create_mock_distribution(samples, bin_count=100):
             """Create a mock Distribution instance with custom bin_count."""
             mock_instance = MagicMock()
             mock_instance.bin_count = bin_count