Skip to content

Commit f19d942

Browse files
Allow empty RaveledVars results in DictToArrayBijection.map
1 parent 36db815 commit f19d942

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

pymc3/blocking.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,11 @@ class DictToArrayBijection:
4141
def map(var_dict: Dict[str, np.ndarray]) -> RaveledVars:
4242
"""Map a dictionary of names and variables to a concatenated 1D array space."""
4343
vars_info = tuple((v, k, v.shape, v.dtype) for k, v in var_dict.items())
44-
res = np.concatenate([v[0].ravel() for v in vars_info])
44+
raveled_vars = [v[0].ravel() for v in vars_info]
45+
if raveled_vars:
46+
res = np.concatenate(raveled_vars)
47+
else:
48+
res = np.array([])
4549
return RaveledVars(res, tuple(v[1:] for v in vars_info))
4650

4751
@staticmethod

0 commit comments

Comments
 (0)