auto detect globals()

This commit is contained in:
Yuyao Huang
2024-04-23 18:11:15 +08:00
parent 3583f02c5b
commit 1b4eb16d51
4 changed files with 58 additions and 28 deletions
+1 -1
View File
@@ -6,7 +6,7 @@ from test_utils import *
def test_torch():
@Commentor("<return>", _globals=globals())
@Commentor("<return>")
def target():
x = torch.ones(4, 5)