From 765dfef74b385532a2f707c20af8407d7f91cf70 Mon Sep 17 00:00:00 2001 From: Caleb Ho Date: Tue, 26 Sep 2023 13:49:47 -0700 Subject: [PATCH] Fix fbcode//multipy/runtime tests (#327) Summary: Pull Request resolved: https://github.com/pytorch/multipy/pull/327 Reviewed By: A-Kokolis Differential Revision: D49651494 fbshipit-source-id: dcc39e0b7c4cb66054fff22c77e21c8abbd3516f --- multipy/runtime/test_deploy.cpp | 2 +- multipy/runtime/test_deploy_gpu.cpp | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/multipy/runtime/test_deploy.cpp b/multipy/runtime/test_deploy.cpp index b559eab7..541e808c 100644 --- a/multipy/runtime/test_deploy.cpp +++ b/multipy/runtime/test_deploy.cpp @@ -511,7 +511,7 @@ TEST(TorchpyTest, TestPyYAML) { torch::deploy::InterpreterManager m(2); auto I = m.acquireOne(); - auto load = I.global("yaml", "load")({kDocument}); + auto load = I.global("yaml", "full_load")({kDocument}); EXPECT_EQ(1, load.attr("__getitem__")({"a"}).toIValue().toInt()); auto dump = I.global("yaml", "dump")({load}); diff --git a/multipy/runtime/test_deploy_gpu.cpp b/multipy/runtime/test_deploy_gpu.cpp index 6759c1fe..3097432d 100644 --- a/multipy/runtime/test_deploy_gpu.cpp +++ b/multipy/runtime/test_deploy_gpu.cpp @@ -133,7 +133,7 @@ TEST(TorchpyTest, TestPyYAML) { torch::deploy::InterpreterManager m(2); auto I = m.acquireOne(); - auto load = I.global("yaml", "load")({kDocument}); + auto load = I.global("yaml", "full_load")({kDocument}); EXPECT_EQ(1, load.attr("__getitem__")({"a"}).toIValue().toInt()); auto dump = I.global("yaml", "dump")({load});