fix: Update test for new Exception syntax

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Apr 9, 2022 · 2357099 · 2357099
1 parent 17340fb
commit 2357099
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/core/conversion/evaluators/test_aten_evaluators.cpp b/tests/core/conversion/evaluators/test_aten_evaluators.cpp
@@ -650,14 +650,15 @@ TEST(Evaluators, AtenFormatRaiseExceptionEvaluatesCorrectly) {
         %2 : str = prim::Constant[value="{} is not equal to {}"]()
         %3 : str = prim::Constant[value="res5_2"]()
         %5713 : Tensor = prim::Uninitialized()
+        %32 : None = prim::Constant()
         %4 : str = aten::format(%2, %1, %3)
         %5 : bool = aten::eq(%1, %3)
         %y : Tensor = prim::If(%5)
             block0():
                 %194 : Tensor = aten::add(%x_1, %x_2, %0)
                 -> (%194)
             block1():
-                prim::RaiseException(%4)
+                prim::RaiseException(%4, %32)
                 -> (%5713)
         return (%y))IR";
   auto g = std::make_shared<torch::jit::Graph>();