diff --git a/pyciemss/mira_integration/compiled_dynamics.py b/pyciemss/mira_integration/compiled_dynamics.py index 806c75b1f..31cccc8aa 100644 --- a/pyciemss/mira_integration/compiled_dynamics.py +++ b/pyciemss/mira_integration/compiled_dynamics.py @@ -50,8 +50,7 @@ def _compile_initial_state_mira( src: mira.modeling.Model, ) -> Callable[..., Tuple[torch.Tensor]]: symbolic_initials = { - get_name(var): src.template_model.initials[get_name(var)].expression.args[0] - for var in src.variables.values() + get_name(var): var.data["expression"].args[0] for var in src.variables.values() } numeric_initial_state_func = sympytorch.SymPyModule( expressions=[symbolic_initials[get_name(var)] for var in src.variables.values()] diff --git a/setup.cfg b/setup.cfg index d8c222fe5..25a911f84 100644 --- a/setup.cfg +++ b/setup.cfg @@ -9,7 +9,7 @@ license_files = LICENSE install_requires = jupyter torch >= 1.8.0 - mira @ git+https://github.com/indralab/mira.git@0.2.0 + mira @ git+https://github.com/indralab/mira.git@0.2.1 chirho @ git+https://github.com/BasisResearch/chirho@f3019d4b22f4e49261efbf8da90a30095af2afbc sympytorch torchdiffeq diff --git a/tests/fixtures.py b/tests/fixtures.py index 140f826bb..061101789 100644 --- a/tests/fixtures.py +++ b/tests/fixtures.py @@ -17,7 +17,7 @@ ] REGNET_URLS = [ - # "https://raw.githubusercontent.com/DARPA-ASKEM/Model-Representations/main/regnet/examples/lotka_volterra.json", + "https://raw.githubusercontent.com/DARPA-ASKEM/Model-Representations/main/regnet/examples/lotka_volterra.json", # "https://raw.githubusercontent.com/DARPA-ASKEM/Model-Representations/main/regnet/examples/syntax_edge_cases.json", ]