From c8e1927964b3b4a25622882daca50f0614be2c6c Mon Sep 17 00:00:00 2001 From: yashbonde Date: Wed, 11 Oct 2023 15:47:01 +0530 Subject: [PATCH] [chore] update cf_internal --- cf_internal | 2 +- chainfury/base.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/cf_internal b/cf_internal index 8e4099cd..83f0c853 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 54166f97..f4852224 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: