search

OpenSourceEconomics / lcm

Solution and simulation of life cycle models in Python with GPU acceleration.
Apache License 2.0
16 stars 1 forks source link

Only works for 2 masks #3

Closed mo2561057 closed 2 years ago

mo2561057 commented 2 years ago

https://github.com/iza-institute-of-labor-economics/lcm/blob/b187c937394c6dcfa7efe083d0ec4e1fe7e208dd/src/lcm/create_state_space.py#L204

When we use jax logical_and we cannot use more than 2 masks. At least it fails when I try to. Can we just use functools reduce or is there a better way?

janosg commented 2 years ago

Fixed. Thanks!