Skip to content

Commit

Permalink
[Torch] Check input_port_id in __eq__ of PTTargetPoint (#2632)
Browse files Browse the repository at this point in the history
### Changes

Check input_port_id in __eq__ of PTTargetPoins

### Reason for changes

To correctly check PTTargetPoints
  • Loading branch information
daniil-lyakhov authored Apr 15, 2024
1 parent 60742a1 commit 4279929
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions nncf/torch/graph/transformations/commands.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 4279929

Please sign in to comment.