From 570320659727c3f6ac6d858e0c87107a21fcd0b4 Mon Sep 17 00:00:00 2001 From: Ashvin Kumar Date: Mon, 18 Nov 2024 09:51:04 -0800 Subject: [PATCH] Fix pylint warnings. Signed-off-by: Ashvin Kumar --- .../torch/src/python/aimet_torch/meta/connectedgraph.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/TrainingExtensions/torch/src/python/aimet_torch/meta/connectedgraph.py b/TrainingExtensions/torch/src/python/aimet_torch/meta/connectedgraph.py index e000829301d..c71b322db0b 100644 --- a/TrainingExtensions/torch/src/python/aimet_torch/meta/connectedgraph.py +++ b/TrainingExtensions/torch/src/python/aimet_torch/meta/connectedgraph.py @@ -157,6 +157,9 @@ def __init__(self, model: torch.nn.Module, model_input: Union[torch.Tensor, Tupl # Counts number of constant inputs there are in the graph self._constant_count = 0 + self._input_structure = None + self._output_structure = None + self._generate_module_lookup_table(model) with in_eval_mode(model), torch.no_grad(): self._aimet_defined_modules = \ @@ -758,10 +761,16 @@ def flatten_pack_producers(deconstruct_op: Op): @property def input_structure(self): + """ + Getter function for input structure + """ return self._input_structure @property def output_structure(self): + """ + Getter function for output structure + """ return self._output_structure def _optimize_connected_graph(self):