Skip to content

Commit d16be7d

Browse files
Fix dispatch types
1 parent 278f625 commit d16be7d

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

symbolic_pymc/meta.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,12 @@
99
from functools import partial
1010
from collections import OrderedDict
1111
from contextlib import contextmanager
12-
from collections.abc import Iterator, Mapping
12+
from collections.abc import Iterator, Mapping, Sequence
1313

1414
from unification import isvar, Var
1515

16+
from etuples.core import ExpressionTuple
17+
1618
from .utils import HashableNDArray
1719

1820
from multipledispatch import dispatch
@@ -74,20 +76,20 @@ def metatize(obj):
7476
return _metatize(obj)
7577

7678

77-
@dispatch((type(None), types.FunctionType, partial, str, dict))
79+
@dispatch((type(None), types.FunctionType, partial, str, Mapping))
7880
def _metatize(obj):
7981
return obj
8082

8183

82-
@_metatize.register((set, tuple))
84+
@_metatize.register((frozenset, tuple, ExpressionTuple))
8385
@cached(metatize_cache)
84-
def _metatize_set_tuple(obj):
86+
def _metatize_hashable_Sequence(obj):
8587
"""Convert elements of an iterable to meta objects."""
8688
return type(obj)([metatize(o) for o in obj])
8789

8890

89-
@_metatize.register(list)
90-
def _metatize_list(obj):
91+
@_metatize.register(Sequence)
92+
def _metatize_Sequence(obj):
9193
"""Convert elements of an iterable to meta objects."""
9294
return type(obj)([metatize(o) for o in obj])
9395

0 commit comments

Comments
 (0)