def METHOD_NAME(self, weights: Dict[str, DT]) -> None:
    """Assign new values to the model's trainable weights.
    Args:
        weights: Model weights, as a dict mapping parameters' names
            to their value.
    """