fix

tigergraph · Nov 18, 2024 · fcac627 · fcac627
1 parent 1a9f542
commit fcac627
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyTigerGraph/gds/dataloaders.py b/pyTigerGraph/gds/dataloaders.py
@@ -37,7 +37,7 @@
 import numpy as np
 import pandas as pd
 
-from ..pyTigerGraphException import TigerGraphException
+from ..common.exception import TigerGraphException
 from .utilities import install_query_file, random_string, add_attribute
 
 __all__ = ["VertexLoader", "EdgeLoader", "NeighborLoader", "GraphLoader", "EdgeNeighborLoader", "NodePieceLoader", "HGTLoader"]

diff --git a/pyTigerGraph/gds/featurizer.py b/pyTigerGraph/gds/featurizer.py
@@ -25,7 +25,7 @@
 if TYPE_CHECKING:
     from ..pyTigerGraph import TigerGraphConnection
 
-from ..pyTigerGraphException import TigerGraphException
+from ..common.exception import TigerGraphException
 import json
 import re
 import time