From 4279929f9f0e7fa9a4f2de43580680a1ba6ae825 Mon Sep 17 00:00:00 2001 From: Daniil Lyakhov Date: Mon, 15 Apr 2024 17:23:02 +0200 Subject: [PATCH] [Torch] Check input_port_id in __eq__ of PTTargetPoint (#2632) ### Changes Check input_port_id in __eq__ of PTTargetPoins ### Reason for changes To correctly check PTTargetPoints --- nncf/torch/graph/transformations/commands.py | 1 + 1 file changed, 1 insertion(+) diff --git a/nncf/torch/graph/transformations/commands.py b/nncf/torch/graph/transformations/commands.py index 4e27236f5e9..b2461277a5f 100644 --- a/nncf/torch/graph/transformations/commands.py +++ b/nncf/torch/graph/transformations/commands.py @@ -57,6 +57,7 @@ def __eq__(self, other: "PTTargetPoint"): isinstance(other, PTTargetPoint) and self.target_type == other.target_type and self.target_node_name == other.target_node_name + and self.input_port_id == other.input_port_id ) def __str__(self):