Source code for prodigy.analysis.evtinvariants.heuristics.templates.templates_factory

from enum import Enum

from prodigy.analysis.evtinvariants.heuristics.templates.polynomial_function import PolynomialEnumeration
from prodigy.analysis.evtinvariants.heuristics.templates.rational_function import RationalFunctionMaxDeg
from prodigy.analysis.evtinvariants.heuristics.templates.templates import TemplateHeuristic



[docs]
class TemplateHeuristics(Enum):
    DEFAULT = RationalFunctionMaxDeg
    POLY = PolynomialEnumeration


[docs]
    @staticmethod
    def create(h_type: 'TemplateHeuristics', *args, **kwargs) -> TemplateHeuristic:
        return h_type.value(*args, **kwargs)