Skip to content

Commit

Permalink
replace pyrsistent.pmap with immutabledict
Browse files Browse the repository at this point in the history
  • Loading branch information
matthiasdiener committed Oct 6, 2023
1 parent f0e277b commit e57f6a6
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions grudge/array_context.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@
from meshmode.array_context import (
PyOpenCLArrayContext as _PyOpenCLArrayContextBase,
PytatoPyOpenCLArrayContext as _PytatoPyOpenCLArrayContextBase)
from pyrsistent import pmap
from immutabledict import immutabledict
from warnings import warn

import logging
Expand Down Expand Up @@ -275,8 +275,8 @@ def _dag_to_compiled_func(self, dict_of_named_arrays,
# }}}

part_id_to_prg = {}
name_in_program_to_tags = pmap()
name_in_program_to_axes = pmap()
name_in_program_to_tags = immutabledict()
name_in_program_to_axes = immutabledict()

from pytato import make_dict_of_named_arrays
for part in distributed_partition.parts.values():
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ def main():
"pymbolic>=2013.2",
"loopy>=2020.2",
"cgen>=2013.1.2",
"dataclasses>=0.7;python_version<='3.6'"
"immutabledict"
],
)

Expand Down

0 comments on commit e57f6a6

Please sign in to comment.