Skip to content

Commit 3a4f4bf

Browse files
calebhofacebook-github-bot
authored andcommitted
buckification
Summary: 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 Differential Revision: D48854790 fbshipit-source-id: e6f361c850bc16df15a2f53200dd55adae907dc3
1 parent 74bd548 commit 3a4f4bf

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)