Fix pyright errors in src/converter/pytorch_converter.py

mlcommons · May 9, 2024 · 4d7e531 · 4d7e531
1 parent 2e1db94
commit 4d7e531
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/converter/pytorch_converter.py b/src/converter/pytorch_converter.py
@@ -81,7 +81,7 @@ def initialize_attributes(self) -> None:
         self.pytorch_time = None
         self.pytorch_start_ts = None
         self.pytorch_finish_ts = None
-        self.pytorch_nodes = None
+        self.pytorch_nodes = dict()
         self.pytorch_root_nids = []
 
         # Initialize node mapping dictionaries