From 2cf92cdad7761713f33382b47e820992650a0842 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Sat, 4 Nov 2023 17:58:19 +0000 Subject: [PATCH] Add missing channel creation --- permchain/pregel/validate.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/permchain/pregel/validate.py b/permchain/pregel/validate.py index 56aa31961..afa51834e 100644 --- a/permchain/pregel/validate.py +++ b/permchain/pregel/validate.py @@ -27,9 +27,14 @@ def validate_chains_channels( channels[chan] = LastValue(Any) if isinstance(input, str): + if input not in channels: + channels[input] = LastValue(Any) if input not in subscribed_channels: raise ValueError(f"Input channel {input} is not subscribed to by any chain") else: + for chan in input: + if chan not in channels: + channels[chan] = LastValue(Any) if all(chan not in subscribed_channels for chan in input): raise ValueError( f"None of the input channels {input} are subscribed to by any chain"