diff --git a/nncf/experimental/torch/fx/node_utils.py b/nncf/experimental/torch/fx/node_utils.py index 96b994cf320..5d03d5e355d 100644 --- a/nncf/experimental/torch/fx/node_utils.py +++ b/nncf/experimental/torch/fx/node_utils.py @@ -12,6 +12,8 @@ import torch +# TODO(dlyakhov): Use torch.fx.graph.find_nodes method instead after +# torch version update (>= 2.4) def get_graph_node_by_name(graph: torch.fx.Graph, name: str) -> torch.fx.Node: """ Retrieves a node with the specified name from the grpah.