satorbis_kit.inference package

class satorbis_kit.inference.BaseInference(config: Dict[str, Any])

Bases: ABC

Base class for inference modules. Provides a standardized template for model inference pipelines, ensuring consistent error handling, and configuration management across different model implementations.

abstractmethod load_model()

Load the model into memory.

abstractmethod postprocess(data: Any) → Any

Transform model output into the desired format.

abstractmethod predict(data: Any) → Any

Execute the model prediction.

abstractmethod preprocess(data: Any) → Any

Prepare the data for model prediction.

abstractmethod validate_input(data: Any) → Any

Validate the input data format and constraints.

abstractmethod validate_output(data: Any) → Any

Validate the final output before returning.

exception satorbis_kit.inference.InferenceError

Bases: Exception

Base exception for all inference-related operations.

exception satorbis_kit.inference.InputValidationError

Bases: InferenceError

exception satorbis_kit.inference.LoadModelError

Bases: InferenceError

exception satorbis_kit.inference.OutputValidationError

Bases: InferenceError

exception satorbis_kit.inference.PostprocessError

Bases: InferenceError

exception satorbis_kit.inference.PredictError

Bases: InferenceError

exception satorbis_kit.inference.PreprocessError

Bases: InferenceError

Submodules

• satorbis_kit.inference.base_inference module
  • BaseInference
    • BaseInference.load_model()
    • BaseInference.postprocess()
    • BaseInference.predict()
    • BaseInference.preprocess()
    • BaseInference.validate_input()
    • BaseInference.validate_output()
• satorbis_kit.inference.exceptions module
  • InferenceError
  • InputValidationError
  • LoadModelError
  • OutputValidationError
  • PostprocessError
  • PredictError
  • PreprocessError