mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
695cb0d342
Fix typo in test folder. _typos.toml ```bash [default.extend-words] nd = "nd" arange = "arange" Nd = "Nd" GLOBALs = "GLOBALs" hte = "hte" iy = "iy" PN = "PN" Dout = "Dout" optin = "optin" gam = "gam" PTD = "PTD" ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/166374 Approved by: https://github.com/cyyever, https://github.com/ezyang
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 = f"{Path(__file__).parent}/package_bc"
|
|
|
|
|
|
class TestLoadBCPackages(PackageTestCase):
|
|
"""Tests for checking loading has backwards compatibility"""
|
|
|
|
@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()
|