base_model

mlte/model/base_model.py

Base model implementation for all MLTE models.

BaseModel

Bases: BaseModel

The base model for all MLTE models.

Source code in mlte/model/base_model.py

class BaseModel(pydantic.BaseModel):
    """The base model for all MLTE models."""

    def to_json(self) -> Dict[str, Any]:
        """
        Serialize the model. Also check if the result is serializable.
        :return: The JSON representation of the model
        """
        json_object = self.model_dump()

        # Check if object can't be serialized.
        try:
            _ = json.dumps(json_object)
        except TypeError as e:
            raise SerializationError(e, str(type(self)))

        return json_object

    @classmethod
    def from_json(cls, data: Dict[str, Any]) -> BaseModel:
        """
        Deserialize a model from data.
        :param data: The raw input data
        :return: A deserialized model instance
        """
        return cls(**data)

    def _equal(a: BaseModel, b: BaseModel) -> bool:
        """
        Compare model instances for equality.

        :param a: Input instance
        :param b: Input instance
        :return: `True` if `a` and `b` are equal, `False` otherwise
        """
        return a.to_json() == b.to_json()

    def __eq__(self, other: object) -> bool:
        """Test instance for equality."""
        if not isinstance(other, BaseModel):
            return False
        return self._equal(other)

__eq__(other)

Test instance for equality.

Source code in mlte/model/base_model.py

def __eq__(self, other: object) -> bool:
    """Test instance for equality."""
    if not isinstance(other, BaseModel):
        return False
    return self._equal(other)

from_json(data) classmethod

Deserialize a model from data.

Parameters:

Name	Type	Description	Default
data	Dict[str, Any]	The raw input data	required

Returns:

Type	Description
BaseModel	A deserialized model instance

Source code in mlte/model/base_model.py

@classmethod
def from_json(cls, data: Dict[str, Any]) -> BaseModel:
    """
    Deserialize a model from data.
    :param data: The raw input data
    :return: A deserialized model instance
    """
    return cls(**data)

to_json()

Serialize the model. Also check if the result is serializable.

Returns:

Type	Description
Dict[str, Any]	The JSON representation of the model

Source code in mlte/model/base_model.py

def to_json(self) -> Dict[str, Any]:
    """
    Serialize the model. Also check if the result is serializable.
    :return: The JSON representation of the model
    """
    json_object = self.model_dump()

    # Check if object can't be serialized.
    try:
        _ = json.dumps(json_object)
    except TypeError as e:
        raise SerializationError(e, str(type(self)))

    return json_object