Update on "Excise yaml dependency in torchgen.model"

zou3519 · zou3519 · commit 63d4e8d543ab · 2023-04-27T14:27:52.000-07:00
The problem: - The new CustomOp API depends on torchgen.model - torchgen.model imports `yaml` - `yaml` is not a PyTorch runtime dependency To unblock myself, because I'm not sure how long it'll take to convince people yaml should be a PyTorch runtime dependency (unless one of you wants to approve #100166), this PR removes the yaml dependency from torchgen.model. It does so by splitting torchgen.utils (the offender) into torchgen.utils (no yaml) and torchgen.yaml (which uses yaml). Test Plan: - CI [ghstack-poisoned]
diff --git a/test/test_meta.py b/test/test_meta.py
@@ -26,7 +26,7 @@
     OpDTypes,
 )
 from torch.testing._internal.common_methods_invocations import op_db
-from torchgen.yaml import YamlLoader
+from torchgen.yaml_utils import YamlLoader
 from torchgen.model import OperatorName
 
 import sys

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@`
`26`	`26`	`OpDTypes,`
`27`	`27`	`)`
`28`	`28`	`from torch.testing._internal.common_methods_invocations import op_db`
`29`		`-from torchgen.yaml import YamlLoader`
	`29`	`+from torchgen.yaml_utils import YamlLoader`
`30`	`30`	`from torchgen.model import OperatorName`
`31`	`31`
`32`	`32`	`import sys`