diff --git a/src/dcegm/likelihood.py b/src/dcegm/likelihood.py index ae1981a3..52a22087 100644 --- a/src/dcegm/likelihood.py +++ b/src/dcegm/likelihood.py @@ -20,15 +20,13 @@ def create_individual_likelihood_function_for_model( model: Dict[str, Any], - options: Dict[str, Any], observed_states: Dict[str, int], observed_wealth: np.array, observed_choices: np.array, - exog_savings_grid: np.ndarray, params_all, ): solve_func = get_solve_func_for_model( - model=model, exog_savings_grid=exog_savings_grid, options=options + model=model, ) observed_state_choice_indexes = get_state_choice_index_per_state( @@ -37,6 +35,8 @@ def create_individual_likelihood_function_for_model( state_space_names=model["model_structure"]["state_space_names"], ) + options = model["options"] + # Create the calculation of the choice probabilities, which takes parameters as # input as well as the solved endogenous wealth grid and the values. def partial_choice_prob_calculation(value_in, endog_grid_in, params_in):