Skip to content

Commit 69722ac

Browse files
calebhofacebook-github-bot
authored andcommitted
buckification (#460)
Summary: Pull Request resolved: #460 This commit was generated using `mgt import`. buckification for third-party libraries: third-party/pypi/pyyaml/6.0.1 third-party/pypi/tblib/1.7.0 third-party/pypi/sagemaker/2.182.0 Reviewed By: luccabb, A-Kokolis Differential Revision: D48854790 fbshipit-source-id: 8f245163b94581a02f2b7e48a8a10f03aaeb5474
1 parent e34ecdd commit 69722ac

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

kernels/test/gen_supported_features_test.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,8 @@ def test_gen_header(self):
3030
dtype_double:
3131
type: bool
3232
default: true
33-
"""
33+
""",
34+
Loader=yaml.FullLoader,
3435
)
3536
result = generate_header(y)
3637
self.assertTrue("bool is_aten = false;" in result)
@@ -45,7 +46,8 @@ def test_gen_def(self):
4546
4647
- namespace: op_gelu
4748
dtype_double: true
48-
"""
49+
""",
50+
Loader=yaml.FullLoader,
4951
)
5052
result = generate_definition(y)
5153
self.assertTrue(".output_resize = true," in result)

0 commit comments

Comments
 (0)