import os

if "DATABRICKS_RUNTIME_VERSION" in os.environ:
    from databricks.sdk.runtime import display as db_display
else:
    db_display = None  # pylint: disable=invalid-name

def df_show(data) -> None:
    """ Wrapper to display dataframes in both, databricks and local environment
    """
    if db_display is not None:
        db_display(data)
    else:
        print(data.show(10))

Which solves the initial question. Now it's to be decided if we should keep logic and visualization separate.