def prepare_fit_kwargs(self) -> None:
        """Prepares fit_kwargs including datamodule using self.config_init['data'] if given."""
        self.fit_kwargs = {"model": self.model}
        if self.datamodule is not None:
            self.fit_kwargs["datamodule"] = self.datamodule
Prepares fit_kwargs including datamodule using self.config_init['data'] if given.
src/flash/core/utilities/lightning_cli.py