From f22d12d2fc227c9b09c65b1f72ef00533148efaf Mon Sep 17 00:00:00 2001 From: fangfengbin <869218239a@zju.edu.cn> Date: Mon, 25 May 2020 15:48:58 +0800 Subject: [PATCH] fix TestGetUncommittedLineage npe bug (#8585) --- src/ray/gcs/redis_gcs_client.h | 2 +- src/ray/raylet/lineage_cache_test.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/ray/gcs/redis_gcs_client.h b/src/ray/gcs/redis_gcs_client.h index 57e3ac515..0d00b1e77 100644 --- a/src/ray/gcs/redis_gcs_client.h +++ b/src/ray/gcs/redis_gcs_client.h @@ -99,7 +99,7 @@ class RAY_EXPORT RedisGcsClient : public GcsClient { HeartbeatBatchTable &heartbeat_batch_table(); DynamicResourceTable &resource_table(); /// Implements the Tasks() interface. - raylet::TaskTable &raylet_task_table(); + virtual raylet::TaskTable &raylet_task_table(); TaskLeaseTable &task_lease_table(); TaskReconstructionLog &task_reconstruction_log(); /// Implements the Errors() interface. diff --git a/src/ray/raylet/lineage_cache_test.cc b/src/ray/raylet/lineage_cache_test.cc index 4bd895d71..196a66fb8 100644 --- a/src/ray/raylet/lineage_cache_test.cc +++ b/src/ray/raylet/lineage_cache_test.cc @@ -157,7 +157,7 @@ class MockGcsClient : public gcs::RedisGcsClient { node_accessor_.reset(new MockNodeInfoAccessor(this, node_id)); } - gcs::raylet::TaskTable &raylet_task_table() { return *task_table_fake_; } + gcs::raylet::TaskTable &raylet_task_table() override { return *task_table_fake_; } MockTaskInfoAccessor &MockTasks() { return *dynamic_cast(task_accessor_.get());