mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
a0e0abd037
This PR fixes a minor typo in `test/jit/test_modules.py`: - Before: `intialized` - After: `initialized` There are no functional code changes — this is a comment-only fix to improve clarity and consistency. Thank you to the PyTorch team for maintaining this outstanding project. Please let me know if anything else is needed. With appreciation, Abhishek Nandy [@abhitorch81](https://github.com/abhitorch81) Pull Request resolved: https://github.com/pytorch/pytorch/pull/157226 Approved by: https://github.com/Skylion007
37 lines
984 B
Python
37 lines
984 B
Python
# Owner(s): ["oncall: jit"]
|
|
|
|
import os
|
|
import sys
|
|
|
|
import torch
|
|
from torch.testing._internal.common_utils import raise_on_run_directly
|
|
from torch.testing._internal.jit_utils import JitTestCase
|
|
|
|
|
|
# Make the helper files in test/ importable
|
|
pytorch_test_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
|
|
sys.path.append(pytorch_test_dir)
|
|
|
|
|
|
class TestModules(JitTestCase):
|
|
def test_script_module_with_constants_list(self):
|
|
"""
|
|
Test that a module that has __constants__ set to something
|
|
that is not a set can be scripted.
|
|
"""
|
|
|
|
# torch.nn.Linear has a __constants__ attribute defined
|
|
# and initialized to a list.
|
|
class Net(torch.nn.Linear):
|
|
x: torch.jit.Final[int]
|
|
|
|
def __init__(self) -> None:
|
|
super().__init__(5, 10)
|
|
self.x = 0
|
|
|
|
self.checkModule(Net(), (torch.randn(5),))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise_on_run_directly("test/test_jit.py")
|