Merge pull request #146 from SongshGeoLab/dev

fix(tracker): 🐛 Rename 'distribution' to 'samples' in AimTracker for API compatibility

Author: SongshGeo

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):

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