Fix a bug in standalone mode (#6179)

8b23ec60 · Da Zheng · GitHub · c6cdeb6b · 8b23ec60
Unverified Commit 8b23ec60 authored Aug 21, 2023 by Da Zheng Committed by GitHub Aug 21, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

python/dgl/distributed/standalone_kvstore.py python/dgl/distributed/standalone_kvstore.py +2 -1

No files found.
--- a/python/dgl/distributed/standalone_kvstore.py
+++ b/python/dgl/distributed/standalone_kvstore.py
@@ -62,7 +62,8 @@ class KVClient(object):
    def delete_data(self, name):
        """delete the data"""
        del self._data[name]
-        self._gdata_name_list.remove(name)
+        if name in self._gdata_name_list:
+            self._gdata_name_list.remove(name)

    def data_name_list(self):
        """get the names of all data"""