From 658e3668ae296be0348f05ed4c8a18ebcf64e226 Mon Sep 17 00:00:00 2001
From: SangBin Cho <rkooo567@gmail.com>
Date: Sat, 7 Aug 2021 18:07:27 -0700
Subject: [PATCH] Fix (#17660)

---
 src/ray/pubsub/publisher.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/ray/pubsub/publisher.cc b/src/ray/pubsub/publisher.cc
index e71d526cc..bd6f6d788 100644
--- a/src/ray/pubsub/publisher.cc
+++ b/src/ray/pubsub/publisher.cc
@@ -95,7 +95,9 @@ bool SubscriptionIndex<KeyIdType>::EraseEntry(const std::string &key_id_binary,
   auto object_it = objects.find(key_id);
   if (object_it == objects.end()) {
     auto it = key_id_to_subscribers_.find(key_id);
-    RAY_CHECK(it->second.count(subscriber_id) == 0);
+    if (it != key_id_to_subscribers_.end()) {
+      RAY_CHECK(it->second.count(subscriber_id) == 0);
+    }
     return false;
   }
   objects.erase(object_it);