Skip to content

Exceptions

This section contains the distilabel custom exceptions. Unlike [errors][../errors.md], exceptions in distilabel are used to handle specific situations that can be anticipated and that can be handled in a controlled way internally by the library.

DistilabelException

Bases: Exception

Base exception (can be gracefully handled) for distilabel framework.

Source code in src/distilabel/exceptions.py 
19
20
21
22
class DistilabelException(Exception):
    """Base exception (can be gracefully handled) for `distilabel` framework."""

    pass

DistilabelGenerationException

Bases: DistilabelException

Base exception for LLM generation errors.

Source code in src/distilabel/exceptions.py 
25
26
27
28
class DistilabelGenerationException(DistilabelException):
    """Base exception for `LLM` generation errors."""

    pass

DistilabelOfflineBatchGenerationNotFinishedException

Bases: DistilabelGenerationException

Exception raised when a batch generation is not finished.

Source code in src/distilabel/exceptions.py 
31
32
33
34
35
36
37
38
39
40
class DistilabelOfflineBatchGenerationNotFinishedException(
    DistilabelGenerationException
):
    """Exception raised when a batch generation is not finished."""

    jobs_ids: Tuple[str, ...]

    def __init__(self, jobs_ids: Tuple[str, ...]) -> None:
        self.jobs_ids = jobs_ids
        super().__init__(f"Batch generation with jobs_ids={jobs_ids} is not finished")