Fix NNAPI deadline overflow test

Bug: 149441015
Test: mma
Test: CtsNNAPITestCases
Change-Id: I1fdb9a5d0ab243b35216c2beeca6893b718cd939

1 files changed, 2 insertions, 2 deletions

diff --git a/nn/runtime/test/TestValidation.cpp b/nn/runtime/test/TestValidation.cpp
index b4204becf..0921f7ab1 100644
--- a/nn/runtime/test/TestValidation.cpp
+++ b/nn/runtime/test/TestValidation.cpp
@@ -2128,7 +2128,7 @@ TEST_F(ValidationTestCompilationForDevices_1, SetTimeoutOverflow) {
         constexpr uint64_t duration = std::numeric_limits<uint64_t>::max();
         EXPECT_EQ(ANeuralNetworksCompilation_setTimeout(mCompilation, duration),
                   ANEURALNETWORKS_NO_ERROR);
-        EXPECT_NE(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_OP_FAILED);
+        EXPECT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_BAD_DATA);
     }
 }
 
@@ -2492,7 +2492,7 @@ void runExecutionSetTimeoutTest(const ANeuralNetworksDevice* device,
                                 n == ANEURALNETWORKS_MISSED_DEADLINE_PERSISTENT);
                     return;
                 case TimeoutDurationType::OVERFLOW:
-                    EXPECT_EQ(n, ANEURALNETWORKS_OP_FAILED);
+                    EXPECT_EQ(n, ANEURALNETWORKS_BAD_DATA);
                     return;
             }
             LOG(FATAL) << "Invalid TimeoutDurationType: " << static_cast<int>(timeoutDurationType);