From bb83d0ca0a41341e1ee10c92c8cb0975370f25c5 Mon Sep 17 00:00:00 2001 From: Romain Date: Tue, 17 Dec 2024 00:33:00 -0800 Subject: [PATCH] Update metaflow/graph.py Make compatible with configs. --- metaflow/graph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metaflow/graph.py b/metaflow/graph.py index ed5f87bb52f..16426006960 100644 --- a/metaflow/graph.py +++ b/metaflow/graph.py @@ -175,7 +175,7 @@ def _create_nodes(self, flow): nodes = {} for element in dir(flow): func = getattr(flow, element) - if hasattr(func, "is_step"): + if callable(func) and hasattr(func, "is_step"): source_file = inspect.getsourcefile(func) source_lines, lineno = inspect.getsourcelines(func) source_code = textwrap.dedent("".join(source_lines))