Source code for nbprint.models.seaborn

from typing import Literal

from IPython.display import HTML
from pydantic import Field

from nbprint import Content

__all__ = ("SeabornDisplayConfiguration",)



[docs]
class SeabornDisplayConfiguration(Content):
    style: Literal["white", "dark", "whitegrid", "darkgrid", "ticks"] = Field(default="whitegrid")
    context: Literal["paper", "notebook", "talk", "poster"] = Field(default="notebook")
    palette: str | list[str] = Field(default="tab10")

    def __call__(self, **_) -> HTML:
        import seaborn as sns

        sns.set_theme(self.context, self.style, self.palette)
        return HTML("")