Skip to content

Commit 74c25ca

Browse files
committed
Fix keep_dot propagation in Driver display functions
1 parent bc7cbbf commit 74c25ca

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

hamilton/driver.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1200,6 +1200,7 @@ def display_downstream_of(
12001200
display_fields=show_schema,
12011201
custom_style_function=custom_style_function,
12021202
config=self.graph._config,
1203+
keep_dot=keep_dot,
12031204
)
12041205
except ImportError as e:
12051206
logger.warning(f"Unable to import {e}", exc_info=True)
@@ -1263,6 +1264,7 @@ def display_upstream_of(
12631264
display_fields=show_schema,
12641265
custom_style_function=custom_style_function,
12651266
config=self.graph._config,
1267+
keep_dot=keep_dot,
12661268
)
12671269
except ImportError as e:
12681270
logger.warning(f"Unable to import {e}", exc_info=True)

0 commit comments

Comments
 (0)