[fix] Resolve issues with metric tracking in SB3 callback (#3186)

aimhubio · Jul 11, 2024 · da429d4 · da429d4
1 parent 86839dc
commit da429d4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -3,9 +3,12 @@
 ## Unreleased
 
 ### Enhancements:
-- Relax `numpy` version upper bound to include `numpy<3`
+- Relax `numpy` version upper bound to include `numpy<3` (judahrand)
 - Add a `-s`/`--skip-if-exists` option to the `init` command to avoid reinitializing a repo if one already exists (stevenjlm)
 
+### Fixes:
+- Fix SB3 callback metric tracking (mihran113)
+
 ## 3.22.0 Jun 20, 2024
 
 ### Enhancements:

diff --git a/aim/sdk/adapters/sb3.py b/aim/sdk/adapters/sb3.py
@@ -28,7 +28,7 @@ def write(
         step: int = 0,
     ) -> None:
         for (key, value), (_, excluded) in zip(sorted(key_values.items()), sorted(key_excluded.items())):
-            if excluded is not None:
+            if excluded is not None and 'aim' in excluded:
                 continue
 
             if isinstance(value, np.ScalarType):