fix bug for self._session_pool (run-llama#7834)

Arize-ai · Sep 26, 2023 · 91f1649 · 91f1649
1 parent d75dd0e
commit 91f1649
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llama_index/graph_stores/nebulagraph.py b/llama_index/graph_stores/nebulagraph.py
@@ -5,7 +5,6 @@
 from typing import Any, Dict, List, Optional
 
 from tenacity import retry, stop_after_attempt, wait_random_exponential
-
 from llama_index.graph_stores.types import GraphStore
 
 QUOTE = '"'
@@ -140,7 +139,8 @@ def __init__(
         self._space_name = space_name
         self._session_pool_kwargs = session_pool_kwargs
 
-        if session_pool is None:
+        self._session_pool: Any = session_pool
+        if self._session_pool is None:
             self.init_session_pool()
 
         self._vid_type = self._get_vid_type()