From 26f1d512185d7ef2519551edd54bb866c542ddb3 Mon Sep 17 00:00:00 2001 From: Atanas Dimitrov Date: Tue, 21 Jan 2025 04:54:54 +0200 Subject: [PATCH] Fix tests Signed-off-by: Atanas Dimitrov --- src/spox/_value_prop_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/spox/_value_prop_backend.py b/src/spox/_value_prop_backend.py index 2fa7a4d..3dc1a28 100644 --- a/src/spox/_value_prop_backend.py +++ b/src/spox/_value_prop_backend.py @@ -40,7 +40,7 @@ def run( self, model: onnx.ModelProto, input_feed: dict[str, RefValue] ) -> dict[str, RefValue]: try: - session = onnx.reference.ReferenceEvaluator(model) + session = onnx.reference.ReferenceEvaluator(model) # type: ignore output_feed = dict(zip(session.output_names, session.run(None, input_feed))) except Exception as e: # Give up on value propagation if an implementation is missing.