frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE][Easy] replace import pathlib with from pathlib import Path (#129426)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/129426
Approved by: https://github.com/malfet
This commit is contained in:
Xuehai Pan
2024-06-30 07:28:38 +08:00
committed by PyTorch MergeBot
parent 2effbcfcd8
commit 4ee1cb9b95
32 changed files with 156 additions and 137 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
test/test_tensorboard.py
View File

@ -1,13 +1,16 @@
# Owner(s): ["module: unknown"]
import expecttest
import io
import numpy as np
import os
import shutil
import sys
import tempfile
import unittest
from pathlib import Path
import expecttest
import numpy as np
TEST_TENSORBOARD = True
try:
@ -36,14 +39,15 @@ skipIfNoMatplotlib = unittest.skipIf(not TEST_MATPLOTLIB, "no matplotlib")
import torch
from torch.testing._internal.common_utils import (
instantiate_parametrized_tests,
IS_MACOS,
IS_WINDOWS,
parametrize,
TestCase,
run_tests,
TEST_WITH_CROSSREF,
IS_WINDOWS,
IS_MACOS,
TestCase,
)
def tensor_N(shape, dtype=float):
numel = np.prod(shape)
x = (np.arange(numel, dtype=dtype)).reshape(shape)
@ -75,15 +79,16 @@ class BaseTestCase(TestCase):
if TEST_TENSORBOARD:
from tensorboard.compat.proto.graph_pb2 import GraphDef
from torch.utils.tensorboard import summary, SummaryWriter
from torch.utils.tensorboard._utils import _prepare_video, convert_to_HWC
from tensorboard.compat.proto.types_pb2 import DataType
from torch.utils.tensorboard.summary import int_to_half, tensor_proto
from torch.utils.tensorboard._convert_np import make_np
from torch.utils.tensorboard._pytorch_graph import graph
from google.protobuf import text_format
from PIL import Image
from tensorboard.compat.proto.graph_pb2 import GraphDef
from tensorboard.compat.proto.types_pb2 import DataType
from torch.utils.tensorboard import summary, SummaryWriter
from torch.utils.tensorboard._convert_np import make_np
from torch.utils.tensorboard._pytorch_graph import graph
from torch.utils.tensorboard._utils import _prepare_video, convert_to_HWC
from torch.utils.tensorboard.summary import int_to_half, tensor_proto
class TestTensorBoardPyTorchNumpy(BaseTestCase):
def test_pytorch_np(self):
@ -289,9 +294,8 @@ class TestTensorBoardSummaryWriter(BaseTestCase):
self.assertTrue(passed)
def test_pathlib(self):
import pathlib
with tempfile.TemporaryDirectory(prefix="test_tensorboard_pathlib") as d:
p = pathlib.Path(d)
p = Path(d)
with SummaryWriter(p) as writer:
writer.add_scalar('test', 1)