From 6d747297e122ec714502549e7c2f1179b5098d01 Mon Sep 17 00:00:00 2001
From: Alex Barghi <105237337+alexbarghi-nv@users.noreply.github.com>
Date: Wed, 1 Nov 2023 12:09:29 -0400
Subject: [PATCH] change exception to TypeError

---
 python/cugraph-pyg/cugraph_pyg/data/cugraph_store.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/cugraph-pyg/cugraph_pyg/data/cugraph_store.py b/python/cugraph-pyg/cugraph_pyg/data/cugraph_store.py
index 9ee6d7c01dc..6192cd621d5 100644
--- a/python/cugraph-pyg/cugraph_pyg/data/cugraph_store.py
+++ b/python/cugraph-pyg/cugraph_pyg/data/cugraph_store.py
@@ -322,7 +322,7 @@ def __del__(self):
             if isinstance(self.__graph._plc_graph, dict):
                 try:
                     distributed.get_client().unpublish_dataset("cugraph_graph")
-                except Exception:
+                except TypeError:
                     warnings.warn(
                         "Could not unpublish graph dataset, most likely because"
                         " dask has already shut down."