Fix fbcode//multipy/runtime tests (#327)

Summary: Pull Request resolved: #327 Reviewed By: A-Kokolis Differential Revision: D49651494 fbshipit-source-id: dcc39e0b7c4cb66054fff22c77e21c8abbd3516f
pytorch · Sep 26, 2023 · 765dfef · 765dfef
1 parent 2460dc6
commit 765dfef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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});