From 7565e8fdada0c5d4805dc664b7f68bf2f1b4374c Mon Sep 17 00:00:00 2001 From: Russell Martin Date: Thu, 19 Dec 2024 11:39:21 -0500 Subject: [PATCH] Export renderers from `pyinstrument.renderers` Type checkers don't like it when you use symbols that are not exported in that module. Putting the renderers in `__all__` will export them here so imports like this one pass mypy: from pyinstrument.renderers import HTMLRenderer Currently one must specify the specific module that owns the renderer to import it. from pyinstrument.renderers.html import HTMLRenderer --- pyinstrument/renderers/__init__.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/pyinstrument/renderers/__init__.py b/pyinstrument/renderers/__init__.py index 98501ea0..60b7f890 100644 --- a/pyinstrument/renderers/__init__.py +++ b/pyinstrument/renderers/__init__.py @@ -5,3 +5,14 @@ from pyinstrument.renderers.pstatsrenderer import PstatsRenderer from pyinstrument.renderers.session import SessionRenderer from pyinstrument.renderers.speedscope import SpeedscopeRenderer + +__all__ = [ + "ConsoleRenderer", + "FrameRenderer", + "HTMLRenderer", + "JSONRenderer", + "PstatsRenderer", + "Renderer", + "SessionRenderer", + "SpeedscopeRenderer", +]