Refactor: Module restructuring, rename ETraceState to HiddenState, add eligibility trace tests, and fix weight computation bugs by chaoming0625 · Pull Request #59 · chaobrain/braintrace

Refactor imports and update module structure: consolidate imports in init.py, rename esd_rtrl to pp_prop, and adjust all exports for better organization

chaoming0625

committed

Refactor version management: move version info to _version.py and update import paths in init.py

chaoming0625

committed

Refactor imports in _rnn.py and base.py: consolidate multi-line imports into single-line imports for improved readability

chaoming0625

committed

Add unit tests for state management functions in _state_management_test.py and refactor imports for improved readability

chaoming0625

committed

Remove LeakySpikeReadout class and update all exports in _readout.py and init.py for better module organization

chaoming0625

committed

Fix weight computation in _etrace_operators.py: remove redundant weight function call in matrix multiplication

chaoming0625

committed

Fix version info typo in _version.py and update all exports in init.py

chaoming0625

committed

Refactor state management and error handling in various modules for improved clarity and robustness

chaoming0625

committed

Refactor module imports for consistency and clarity across the codebase

chaoming0625

committed

Remove LeakySpikeReadout tests for better code clarity and maintenance

chaoming0625

committed

Refactor import statements in multiple files for improved clarity and consistency

chaoming0625

committed

Refactor import statements in multiple files for improved clarity and consistency

chaoming0625

committed

Add unit tests for ETraceAlgorithm and EligibilityTrace functionality

chaoming0625

committed

Refactor test files and add new tests for hybrid algorithms

chaoming0625

committed

Enhance hybrid_test.py with JAX integration and additional unit tests for gradient checks

chaoming0625

committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Refactor: Module restructuring, rename ETraceState to HiddenState, add eligibility trace tests, and fix weight computation bugs#59

Refactor: Module restructuring, rename ETraceState to HiddenState, add eligibility trace tests, and fix weight computation bugs#59
chaoming0625 merged 19 commits intomainfrom
update

Commits on Mar 17, 2026

Refactor imports and update module structure: consolidate imports in init.py, rename esd_rtrl to pp_prop, and adjust all exports for better organization

Refactor version management: move version info to _version.py and update import paths in init.py

Refactor imports in _rnn.py and base.py: consolidate multi-line imports into single-line imports for improved readability

Add unit tests for state management functions in _state_management_test.py and refactor imports for improved readability

Remove LeakySpikeReadout class and update all exports in _readout.py and init.py for better module organization

Fix weight computation in _etrace_operators.py: remove redundant weight function call in matrix multiplication

Fix version info typo in _version.py and update all exports in init.py

Refactor state management and error handling in various modules for improved clarity and robustness

Refactor module imports for consistency and clarity across the codebase

Remove LeakySpikeReadout tests for better code clarity and maintenance

Refactor import statements in multiple files for improved clarity and consistency

Refactor import statements in multiple files for improved clarity and consistency

Add unit tests for ETraceAlgorithm and EligibilityTrace functionality

Refactor test files and add new tests for hybrid algorithms

Enhance hybrid_test.py with JAX integration and additional unit tests for gradient checks

Commits on Mar 18, 2026

Fix tuple syntax in weight initialization across ETrace model tests

Update documentation to reflect class renaming from ETraceState to HiddenState and related changes

Refactor code for improved readability by removing unused imports and formatting initialization parameters

Update base_test.py to accommodate JAX 0.9.0+ changes by renaming 'effects' to 'effs' in JaxprEqn initialization