Skip to content

model_handler

Classes

ModelHandler

ModelHandler(context: CodeGenContext)

Responsible for generating model source code.

Functions

compute_values
compute_values() -> dict[str, str]

Computes the values needed for generating the model source code.

Returns:

  • dict ( dict[str, str] ) –

    A dictionary containing the computed values for code generation.

emit
emit(save_path: Path)

Generate the source code for the main inference model.

Parameters:

  • save_path
    (Path) –

    Path to save the generated code.

Functions