diff --git a/src/snowflake/snowpark/lineage.py b/src/snowflake/snowpark/lineage.py index bf2b421478b..68a1236d2f1 100644 --- a/src/snowflake/snowpark/lineage.py +++ b/src/snowflake/snowpark/lineage.py @@ -20,6 +20,7 @@ _MIN_TRACE_DEPTH = 1 _MAX_TRACE_DEPTH = 5 +_DEFAULT_TRACE_DEPTH = 2 class LineageDirection(Enum): @@ -429,7 +430,7 @@ def trace( *, object_version: Optional[str] = None, direction: Union[str, LineageDirection] = LineageDirection.BOTH, - depth: int = 2, + depth: int = _DEFAULT_TRACE_DEPTH, ) -> "snowflake.snowpark.dataframe.DataFrame": """ Traces the lineage of an object within Snowflake and returns it as a DataFrame.