OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Define __all__ for torch.utils.tensorboard (#147550)

Browse Source 
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: https://github.com/pytorch/pytorch/pull/147550
Approved by: https://github.com/albanD
This commit is contained in:
Matthew Hoffman 2025-02-28 23:06:06 +00:00 committed by PyTorch MergeBot
parent 3a69dee955
commit 1ae7cc41ca
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
torch/utils/tensorboard/__init__.py
View File

@ -9,5 +9,11 @@ if not hasattr(tensorboard, "__version__") or Version(
del Version
del tensorboard
from .writer import FileWriter, SummaryWriter # noqa: F401
from tensorboard.summary.writer.record_writer import RecordWriter # noqa: F401
from .writer import FileWriter, SummaryWriter
from tensorboard.summary.writer.record_writer import RecordWriter
__all__ = [
"FileWriter",
"RecordWriter",
"SummaryWriter",
]