Skip to content

Commit 1ae7cc4

Browse files
ringohoffmanpytorchmergebot
authored andcommitted
Define __all__ for torch.utils.tensorboard (pytorch#147550)
Fixes the issue: ```python import torch.utils.tensorboard torch.utils.tensorboard.FileWriter # pyright: "FileWriter" is not exported from module "torch.utils.tensorboard" torch.utils.tensorboard.RecordWriter # pyright: "RecordWriter" is not exported from module "torch.utils.tensorboard" torch.utils.tensorboard.SummaryWriter # pyright: "SummaryWriter" is not exported from module "torch.utils.tensorboard" ``` The [docs page for `torch.utils.tensorboard`](https://pytorch.org/docs/stable/tensorboard.html) Pull Request resolved: pytorch#147550 Approved by: https://github.com/albanD
1 parent 3a69dee commit 1ae7cc4

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

torch/utils/tensorboard/__init__.py

+8-2
Original file line numberDiff line numberDiff line change
@@ -9,5 +9,11 @@
99
del Version
1010
del tensorboard
1111

12-
from .writer import FileWriter, SummaryWriter # noqa: F401
13-
from tensorboard.summary.writer.record_writer import RecordWriter # noqa: F401
12+
from .writer import FileWriter, SummaryWriter
13+
from tensorboard.summary.writer.record_writer import RecordWriter
14+
15+
__all__ = [
16+
"FileWriter",
17+
"RecordWriter",
18+
"SummaryWriter",
19+
]

0 commit comments

Comments
 (0)