import get_module from utils instead of lineage._utils

jkasiraj · jkasiraj · commit 2d24e9f51358 · 2023-04-28T11:43:14.000-07:00
diff --git a/src/sagemaker/lineage/query.py b/src/sagemaker/lineage/query.py
@@ -18,7 +18,8 @@
 from typing import Optional, Union, List, Dict
 import re
 
-from sagemaker.lineage._utils import get_resource_name_from_arn, get_module
+from sagemaker.utils import get_module
+from sagemaker.lineage._utils import get_resource_name_from_arn
 
 
 class LineageEntityEnum(Enum):
