diff --git a/cf_internal b/cf_internal index 8e4099c..83f0c85 160000 --- a/cf_internal +++ b/cf_internal @@ -1 +1 @@ -Subproject commit 8e4099cd4aac5fd85200633767bae2f0fe8c73c1 +Subproject commit 83f0c853d52aaf0f34d4a8fbddc3e8abc6e0fcae diff --git a/chainfury/base.py b/chainfury/base.py index 54166f9..f485222 100644 --- a/chainfury/base.py +++ b/chainfury/base.py @@ -948,6 +948,8 @@ def __init__( self.topo_order = topological_sort(self.edges) self.sample = sample self.main_in = main_in + + op_key = main_out.split("/")[-1] # TODO: @yashbonde - make this useful self.main_out = main_out for node_id in self.topo_order: