mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
45411d1fc9
Changes: 1. Always explicit `.absolute()`: `Path(__file__)` -> `Path(__file__).absolute()` 2. Replace `path.resolve()` with `path.absolute()` if the code is resolving the PyTorch repo root directory. Pull Request resolved: https://github.com/pytorch/pytorch/pull/129409 Approved by: https://github.com/albanD
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
# Owner(s): ["oncall: package/deploy"]
|
|
|
|
from pathlib import Path
|
|
from unittest import skipIf
|
|
|
|
from torch.package import PackageImporter
|
|
from torch.testing._internal.common_utils import IS_FBCODE, IS_SANDCASTLE, run_tests
|
|
|
|
|
|
try:
|
|
from .common import PackageTestCase
|
|
except ImportError:
|
|
# Support the case where we run this file directly.
|
|
from common import PackageTestCase
|
|
|
|
packaging_directory = str(Path(__file__).absolute().parent / "package_bc")
|
|
|
|
|
|
class TestLoadBCPackages(PackageTestCase):
|
|
"""Tests for checking loading has backwards compatiblity"""
|
|
|
|
@skipIf(
|
|
IS_FBCODE or IS_SANDCASTLE,
|
|
"Tests that use temporary files are disabled in fbcode",
|
|
)
|
|
def test_load_bc_packages_nn_module(self):
|
|
"""Tests for backwards compatible nn module"""
|
|
importer1 = PackageImporter(f"{packaging_directory}/test_nn_module.pt")
|
|
importer1.load_pickle("nn_module", "nn_module.pkl")
|
|
|
|
@skipIf(
|
|
IS_FBCODE or IS_SANDCASTLE,
|
|
"Tests that use temporary files are disabled in fbcode",
|
|
)
|
|
def test_load_bc_packages_torchscript_module(self):
|
|
"""Tests for backwards compatible torchscript module"""
|
|
importer2 = PackageImporter(f"{packaging_directory}/test_torchscript_module.pt")
|
|
importer2.load_pickle("torchscript_module", "torchscript_module.pkl")
|
|
|
|
@skipIf(
|
|
IS_FBCODE or IS_SANDCASTLE,
|
|
"Tests that use temporary files are disabled in fbcode",
|
|
)
|
|
def test_load_bc_packages_fx_module(self):
|
|
"""Tests for backwards compatible fx module"""
|
|
importer3 = PackageImporter(f"{packaging_directory}/test_fx_module.pt")
|
|
importer3.load_pickle("fx_module", "fx_module.pkl")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_tests()
|