making default a macro

snowflakedb · Apr 29, 2024 · 25e2da0 · 25e2da0
1 parent 8aac228
commit 25e2da0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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.