@@ -791,6 +791,7 @@ def _visualize_execution_helper(
791
791
deduplicate_inputs : bool = False ,
792
792
show_schema : bool = True ,
793
793
custom_style_function : Callable = None ,
794
+ bypass_validation : bool = False ,
794
795
):
795
796
"""Helper function to visualize execution, using a passed-in function graph.
796
797
@@ -809,7 +810,8 @@ def _visualize_execution_helper(
809
810
"""
810
811
# TODO should determine if the visualization logic should live here or in the graph.py module
811
812
nodes , user_nodes = fn_graph .get_upstream_nodes (final_vars , inputs , overrides )
812
- Driver .validate_inputs (fn_graph , adapter , user_nodes , inputs , nodes )
813
+ if not bypass_validation :
814
+ Driver .validate_inputs (fn_graph , adapter , user_nodes , inputs , nodes )
813
815
node_modifiers = {fv : {graph .VisualizationNodeModifiers .IS_OUTPUT } for fv in final_vars }
814
816
for user_node in user_nodes :
815
817
if user_node .name not in node_modifiers :
@@ -857,6 +859,7 @@ def visualize_execution(
857
859
deduplicate_inputs : bool = False ,
858
860
show_schema : bool = True ,
859
861
custom_style_function : Callable = None ,
862
+ bypass_validation : bool = False ,
860
863
) -> Optional ["graphviz.Digraph" ]: # noqa F821
861
864
"""Visualizes Execution.
862
865
@@ -907,6 +910,7 @@ def visualize_execution(
907
910
deduplicate_inputs = deduplicate_inputs ,
908
911
show_schema = show_schema ,
909
912
custom_style_function = custom_style_function ,
913
+ bypass_validation = bypass_validation ,
910
914
)
911
915
912
916
@capture_function_usage
@@ -1500,6 +1504,7 @@ def visualize_materialization(
1500
1504
deduplicate_inputs : bool = False ,
1501
1505
show_schema : bool = True ,
1502
1506
custom_style_function : Callable = None ,
1507
+ bypass_validation : bool = False ,
1503
1508
) -> Optional ["graphviz.Digraph" ]: # noqa F821
1504
1509
"""Visualizes materialization. This helps give you a sense of how materialization
1505
1510
will impact the DAG.
@@ -1521,6 +1526,7 @@ def visualize_materialization(
1521
1526
:param show_schema: If True, show the schema of the materialized nodes
1522
1527
if nodes have schema metadata attached.
1523
1528
:param custom_style_function: Optional. Custom style function.
1529
+ :param bypass_validation: If True, bypass validation. Optional.
1524
1530
:return: The graphviz graph, if you want to do something with it
1525
1531
"""
1526
1532
if additional_vars is None :
@@ -1547,6 +1553,7 @@ def visualize_materialization(
1547
1553
deduplicate_inputs = deduplicate_inputs ,
1548
1554
show_schema = show_schema ,
1549
1555
custom_style_function = custom_style_function ,
1556
+ bypass_validation = bypass_validation ,
1550
1557
)
1551
1558
1552
1559
def validate_execution (
0 commit comments