alberta-framework 0.3.0__tar.gz → 0.3.2__tar.gz

{alberta_framework-0.3.0 → alberta_framework-0.3.2}/CLAUDE.md

@@ -470,6 +470,11 @@ The publish workflow uses OpenID Connect (no API tokens). Configure on PyPI:
 
 ## Changelog
 
+### v0.3.2 (2026-02-03)
+- **FIX**: Relaxed test tolerance in batched vs sequential comparison tests (`rtol=1e-5`) to account for floating-point differences between vmap and sequential execution paths
+- **FIX**: Added `ignore = ["F722"]` to ruff config for jaxtyping shape annotation syntax that ruff doesn't understand
+- **FIX**: Removed unused `PRNGKeyArray` import from `core/types.py`
+
 ### v0.3.0 (2026-02-03)
 - **FEATURE**: Migrated all state types from NamedTuple to `@chex.dataclass(frozen=True)` for DeepMind-style JAX compatibility
 - **FEATURE**: Added jaxtyping shape annotations for compile-time type safety (`Float[Array, " feature_dim"]`, `PRNGKeyArray`, etc.)

{alberta_framework-0.3.0 → alberta_framework-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alberta-framework
-Version: 0.3.0
+Version: 0.3.2
 Summary: Implementation of the Alberta Plan for AI Research - continual learning with meta-learned step-sizes
 Project-URL: Homepage, https://github.com/j-klawson/alberta-framework
 Project-URL: Repository, https://github.com/j-klawson/alberta-framework

{alberta_framework-0.3.0 → alberta_framework-0.3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "alberta-framework"
-version = "0.3.0"
+version = "0.3.2"
 description = "Implementation of the Alberta Plan for AI Research - continual learning with meta-learned step-sizes"
 readme = "README.md"
 license = "Apache-2.0"
@@ -71,6 +71,8 @@ target-version = "py313"
 
 [tool.ruff.lint]
 select = ["E", "F", "I", "N", "W", "UP"]
+# F722: Syntax error in forward annotation - ruff doesn't understand jaxtyping shape annotations
+ignore = ["F722"]
 
 [tool.mypy]
 python_version = "3.13"

{alberta_framework-0.3.0 → alberta_framework-0.3.2}/src/alberta_framework/core/types.py

@@ -9,7 +9,7 @@ from typing import TYPE_CHECKING
 import chex
 import jax.numpy as jnp
 from jax import Array
-from jaxtyping import Float, Int, PRNGKeyArray
+from jaxtyping import Float, Int
 
 if TYPE_CHECKING:
     from alberta_framework.core.learners import NormalizedLearnerState

{alberta_framework-0.3.0 → alberta_framework-0.3.2}/tests/test_learners.py

@@ -576,8 +576,8 @@ class TestBatchedLearningLoop:
             sequential_metrics.append(metrics)
         sequential_metrics = jnp.stack(sequential_metrics)
 
-        # Should match
-        chex.assert_trees_all_close(batched_result.metrics, sequential_metrics)
+        # Should match (use rtol=1e-5 to account for vmap vs sequential floating-point differences)
+        chex.assert_trees_all_close(batched_result.metrics, sequential_metrics, rtol=1e-5)
 
     def test_batched_with_step_size_tracking(self, rng_key):
         """Batched loop should support step-size tracking."""
@@ -704,8 +704,8 @@ class TestBatchedNormalizedLearningLoop:
             sequential_metrics.append(metrics)
         sequential_metrics = jnp.stack(sequential_metrics)
 
-        # Should match
-        chex.assert_trees_all_close(batched_result.metrics, sequential_metrics)
+        # Should match (use rtol=1e-5 to account for vmap vs sequential floating-point differences)
+        chex.assert_trees_all_close(batched_result.metrics, sequential_metrics, rtol=1e-5)
 
     def test_normalized_batched_with_both_tracking(self, rng_key):
         """Batched normalized loop should support both tracking options."""