dcegm.pre_processing.shared

Functions

determine_function_arguments_and_partial_model_specs(...)
partial_model_specs_and_update_signature(func, ...)	Partial in model_specs and update signature.
create_array_with_smallest_int_dtype(arr)	Return array with the smallest unsigned integer dtype.
get_smallest_int_type(n_values)	Return the smallest unsigned integer type that can hold n_values.
try_jax_array(x)

Module Contents

dcegm.pre_processing.shared.determine_function_arguments_and_partial_model_specs(func, model_specs, not_allowed_state_choices=None, continuous_state_name=None)

dcegm.pre_processing.shared.partial_model_specs_and_update_signature(func, signature, model_specs)

Partial in model_specs and update signature.

dcegm.pre_processing.shared.create_array_with_smallest_int_dtype(arr)

Return array with the smallest unsigned integer dtype.

dcegm.pre_processing.shared.get_smallest_int_type(n_values)

Return the smallest unsigned integer type that can hold n_values.

dcegm.pre_processing.shared.try_jax_array(x)