From da6604681bdc688251d3199d5402cd0ded090fb2 Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Tue, 10 Dec 2024 20:11:35 +0000 Subject: [PATCH] Auto-format by https://ultralytics.com/actions --- onnxslim/third_party/onnx_graphsurgeon/ir/node.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxslim/third_party/onnx_graphsurgeon/ir/node.py b/onnxslim/third_party/onnx_graphsurgeon/ir/node.py index a3b23a1..e87fe76 100644 --- a/onnxslim/third_party/onnx_graphsurgeon/ir/node.py +++ b/onnxslim/third_party/onnx_graphsurgeon/ir/node.py @@ -19,7 +19,7 @@ from dataclasses import dataclass from typing import Dict, List, Union -from onnxslim.third_party.onnx_graphsurgeon.ir.tensor import Constant, Tensor, Variable +from onnxslim.third_party.onnx_graphsurgeon.ir.tensor import Constant, Tensor from onnxslim.third_party.onnx_graphsurgeon.logger import G_LOGGER from onnxslim.third_party.onnx_graphsurgeon.util import misc @@ -238,7 +238,7 @@ def feeds(self): feeds.extend(input if feed.op == "Split" else feed for feed in input.inputs) return feeds - def replace_all_uses_with(self, node: Union['Node', 'Tensor'], input_var_idx=0, output_var_idx=0): + def replace_all_uses_with(self, node: Union["Node", "Tensor"], input_var_idx=0, output_var_idx=0): """Replace all uses of this node with the given node.""" if isinstance(node, Node): input_var = node.outputs[output_var_idx]