Automated Code Change

PiperOrigin-RevId: 621066820
tensorflow · Apr 2, 2024 · 8ceb574 · 8ceb574
1 parent 233305a
commit 8ceb574
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/third_party/xla/xla/python/ifrt/mock.h b/third_party/xla/xla/python/ifrt/mock.h
@@ -66,7 +66,7 @@ class MockArray final : public llvm::RTTIExtends<MockArray, Array> {
               (const, final));
   MOCK_METHOD(absl::StatusOr<std::unique_ptr<PjRtLayout>>, layout, (),
               (const, final));
-  MOCK_METHOD(StatusOr<std::vector<tsl::RCReference<Array>>>,
+  MOCK_METHOD(absl::StatusOr<std::vector<tsl::RCReference<Array>>>,
               DisassembleIntoSingleDeviceArrays, (ArrayCopySemantics semantics),
               (final));
   MOCK_METHOD(absl::StatusOr<tsl::RCReference<Array>>, FullyReplicatedShard,