def __compile_functions__(self): from dolang.function_compiler import make_method_from_factory from dolang.vectorize import standard_function from dolo.compiler.factories import get_factory from .misc import LoosyDict equivalent_function_names = { 'equilibrium': 'arbitrage', 'optimality': 'arbitrage' } functions = LoosyDict(equivalences=equivalent_function_names) original_functions = {} original_gufunctions = {} funnames = [*self.equations.keys()] + ['auxiliary'] import dolo.config debug = dolo.config.debug for funname in funnames: fff = get_factory(self, funname) fun, gufun = make_method_from_factory(fff, vectorize=True, debug=debug) n_output = len(fff.content) functions[funname] = standard_function(gufun, n_output) original_gufunctions[funname] = gufun # basic gufun function original_functions[funname] = fun # basic numba fun self.__original_functions__ = original_functions self.__original_gufunctions__ = original_gufunctions self.__functions__ = functions
def __compile_functions__(self): from dolang.function_compiler import make_method_from_factory from dolang.vectorize import standard_function from dolo.compiler.factories import get_factory functions = {} original_functions = {} original_gufunctions = {} funnames = [*self.equations.keys()] + ['auxiliary'] import dolo.config debug = dolo.config.debug for funname in funnames: fff = get_factory(self, funname) fun, gufun = make_method_from_factory(fff, vectorize=True, debug=debug) n_output = len(fff.content) functions[funname] = standard_function(gufun, n_output) original_gufunctions[funname] = gufun # basic gufun function original_functions[funname] = fun # basic numba fun self.__original_functions__ = original_functions self.__original_gufunctions__ = original_gufunctions self.__functions__ = functions
def __compile_functions__(self): from dolang.function_compiler import make_method_from_factory from dolang.vectorize import standard_function from dolo.compiler.factories import get_factory functions = {} original_functions = {} original_gufunctions = {} funnames = [*self.equations.keys()] + ['auxiliary'] import dolo.config debug = dolo.config.debug for funname in funnames: fff = get_factory(self, funname) fun, gufun = make_method_from_factory( fff, vectorize=True, debug=debug) n_output = len(fff.content) functions[funname] = standard_function(gufun, n_output) original_gufunctions[funname] = gufun # basic gufun function original_functions[funname] = fun # basic numba fun self.__original_functions__ = original_functions self.__original_gufunctions__ = original_gufunctions self.functions = functions