From b11d761e1546903d445abcffdc2866d3553206ba Mon Sep 17 00:00:00 2001 From: Atanas Dimitrov Date: Wed, 15 Jan 2025 13:32:20 +0200 Subject: [PATCH] Prototype Signed-off-by: Atanas Dimitrov --- src/spox/_inline.py | 2 +- src/spox/_value_prop_backend.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/spox/_inline.py b/src/spox/_inline.py index ee5e6da..55c5ad1 100644 --- a/src/spox/_inline.py +++ b/src/spox/_inline.py @@ -142,7 +142,7 @@ def propagate_values( ): return {} input_feed = { - i.name: value_prop_backend.wrap_feed(input_prop_values.get(i.name)) # type: ignore + i.name: value_prop_backend.wrap_feed(input_prop_values[i.name]) for i in self.model.graph.input if i.name in input_prop_values } diff --git a/src/spox/_value_prop_backend.py b/src/spox/_value_prop_backend.py index 0103fd8..2cfd617 100644 --- a/src/spox/_value_prop_backend.py +++ b/src/spox/_value_prop_backend.py @@ -1,4 +1,4 @@ -# Copyright (c) QuantCo 2023-2024 +# Copyright (c) QuantCo 2023-2025 # SPDX-License-Identifier: BSD-3-Clause from __future__ import annotations