alberta-framework 0.2.0__tar.gz → 0.2.1__tar.gz

{alberta_framework-0.2.0 → alberta_framework-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alberta-framework
-Version: 0.2.0
+Version: 0.2.1
 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.2.0 → alberta_framework-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "alberta-framework"
-version = "0.2.0"
+version = "0.2.1"
 description = "Implementation of the Alberta Plan for AI Research - continual learning with meta-learned step-sizes"
 readme = "README.md"
 license = "Apache-2.0"

{alberta_framework-0.2.0 → alberta_framework-0.2.1}/src/alberta_framework/core/types.py

@@ -4,11 +4,14 @@ This module defines the core data types used throughout the framework,
 following JAX conventions with immutable NamedTuples for state management.
 """
 
-from typing import NamedTuple
+from typing import TYPE_CHECKING, NamedTuple
 
 import jax.numpy as jnp
 from jax import Array
 
+if TYPE_CHECKING:
+    from alberta_framework.core.learners import NormalizedLearnerState
+
 # Type aliases for clarity
 Observation = Array  # x_t: feature vector
 Target = Array  # y*_t: desired output