Inspectors

Common abstractions for inspectors

TesterSignatureSpec

Bases: BaseModel

Source code in src/fed_rag/inspectors/common.py

class TesterSignatureSpec(BaseModel):
    __test__ = (
        False  # needed for Pytest collision. Avoids PytestCollectionWarning
    )
    net_parameter: str
    test_data_param: str
    extra_test_kwargs: list[str] = []
    net_parameter_class_name: str

TesterSignatureSpec

Bases: BaseModel

Source code in src/fed_rag/inspectors/common.py

class TesterSignatureSpec(BaseModel):
    __test__ = (
        False  # needed for Pytest collision. Avoids PytestCollectionWarning
    )
    net_parameter: str
    test_data_param: str
    extra_test_kwargs: list[str] = []
    net_parameter_class_name: str

TesterResult

TrainResult

Bases: BaseModel

Source code in src/fed_rag/types/results.py

class TrainResult(BaseModel):
    loss: float

TestResult

Bases: BaseModel

Source code in src/fed_rag/types/results.py

class TestResult(BaseModel):
    __test__ = (
        False  # needed for Pytest collision. Avoids PytestCollectionWarning
    )
    loss: float
    metrics: dict[str, Any] = Field(
        description="Additional metrics computed on test set.",
        default_factory=dict,
    )