xref: /aosp_15_r20/external/pytorch/test/test_package.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker# Owner(s): ["oncall: package/deploy"]
2*da0073e9SAndroid Build Coastguard Worker
3*da0073e9SAndroid Build Coastguard Workerfrom package.package_a.test_all_leaf_modules_tracer import (  # noqa: F401
4*da0073e9SAndroid Build Coastguard Worker    TestAllLeafModulesTracer,
5*da0073e9SAndroid Build Coastguard Worker)
6*da0073e9SAndroid Build Coastguard Workerfrom package.package_a.test_nn_module import TestNnModule  # noqa: F401
7*da0073e9SAndroid Build Coastguard Workerfrom package.test_analyze import TestAnalyze  # noqa: F401
8*da0073e9SAndroid Build Coastguard Workerfrom package.test_dependency_api import TestDependencyAPI  # noqa: F401
9*da0073e9SAndroid Build Coastguard Workerfrom package.test_dependency_hooks import TestDependencyHooks  # noqa: F401
10*da0073e9SAndroid Build Coastguard Workerfrom package.test_digraph import TestDiGraph  # noqa: F401
11*da0073e9SAndroid Build Coastguard Workerfrom package.test_directory_reader import DirectoryReaderTest  # noqa: F401
12*da0073e9SAndroid Build Coastguard Workerfrom package.test_glob_group import TestGlobGroup  # noqa: F401
13*da0073e9SAndroid Build Coastguard Workerfrom package.test_importer import TestImporter  # noqa: F401
14*da0073e9SAndroid Build Coastguard Workerfrom package.test_load_bc_packages import TestLoadBCPackages  # noqa: F401
15*da0073e9SAndroid Build Coastguard Workerfrom package.test_mangling import TestMangling  # noqa: F401
16*da0073e9SAndroid Build Coastguard Workerfrom package.test_misc import TestMisc  # noqa: F401
17*da0073e9SAndroid Build Coastguard Workerfrom package.test_model import ModelTest  # noqa: F401
18*da0073e9SAndroid Build Coastguard Workerfrom package.test_package_fx import TestPackageFX  # noqa: F401
19*da0073e9SAndroid Build Coastguard Workerfrom package.test_package_script import TestPackageScript  # noqa: F401
20*da0073e9SAndroid Build Coastguard Workerfrom package.test_repackage import TestRepackage  # noqa: F401
21*da0073e9SAndroid Build Coastguard Workerfrom package.test_resources import TestResources  # noqa: F401
22*da0073e9SAndroid Build Coastguard Workerfrom package.test_save_load import TestSaveLoad  # noqa: F401
23*da0073e9SAndroid Build Coastguard Worker
24*da0073e9SAndroid Build Coastguard Worker
25*da0073e9SAndroid Build Coastguard Workerif __name__ == "__main__":
26*da0073e9SAndroid Build Coastguard Worker    from torch.testing._internal.common_utils import run_tests
27*da0073e9SAndroid Build Coastguard Worker
28*da0073e9SAndroid Build Coastguard Worker    run_tests()
29