Format position_id_patch.py to pass lint

Author: abeeha123

position_id_patch.py

@@ -5,8 +5,10 @@
 from torch.export import export as torch_export
 from tvm.relax.frontend.torch import from_exported_program
 
+
 class StateDictWrapper(dict):
     """Wrap exported state_dict and inject extra keys (non-persistent buffers)."""
+
     def __init__(self, base_dict, extra):
         super().__init__(base_dict)
         self.extra = extra
@@ -21,6 +23,7 @@ def get(self, key, default=None):
             return self.extra[key]
         return super().get(key, default)
 
+
 class M(nn.Module):
     def __init__(self):
         super().__init__()
@@ -31,6 +34,7 @@ def forward(self, x, mask=None):
         out = self.bert(x, attention_mask=mask).last_hidden_state[:, 0, :]
         return self.cls(out)
 
+
 def main():
     torch.manual_seed(0)
     m = M().eval()
@@ -72,7 +76,9 @@ def __getattr__(self, name):
     except Exception as e:
         print("\n TVM import failed with exception:")
         import traceback
+
         traceback.print_exc()
 
+
 if __name__ == "__main__":
     main()