search

econ-ark / HARK

Heterogenous Agents Resources & toolKit
Apache License 2.0
336 stars 199 forks source link

expected breaks on functions that involve manipulation of arrays #1488

Open sbenthall opened 3 months ago

sbenthall commented 3 months ago 
from HARK.distribution import expected, Normal, DiscreteDistributionLabeled
import numpy as np

gamma = DiscreteDistributionLabeled.from_unlabeled(
    Normal(mu= 0, sigma = 1).discretize(**{"N" : 7}), var_names=["gamma"]
)

expected(func=lambda x: x['gamma'] * np.array([1,2]), dist=gamma)

gets this error:

ValueError: operands could not be broadcast together with shapes (7,) (2,)