def test_comet_log_graph(comet_mock, tmp_path, monkeypatch):
    """Test that CometLogger.log_hyperparams calls internal API method."""
    _patch_comet_atexit(monkeypatch)

    logger = CometLogger(project_name="test")
    model = Mock()

    logger.log_graph(model=model)

    logger.experiment.__internal_api__set_model_graph__.assert_called_once_with(
        graph=model,
        framework="pytorch-lightning",
    )
Test that CometLogger.log_hyperparams calls internal API method.
tests/tests_pytorch/loggers/test_comet.py