Fix lint issues

pytorch · Aug 7, 2023 · 49fb3f6 · 49fb3f6
1 parent 364c20d
commit 49fb3f6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/torch/mps/__init__.py b/torch/mps/__init__.py
@@ -125,6 +125,6 @@ def driver_allocated_memory() -> int:
     "set_per_process_memory_fraction",
     "current_allocated_memory",
     "driver_allocated_memory",
-	"Event",
+    "Event",
     "profiler",
 ]
diff --git a/torch/mps/event.py b/torch/mps/event.py
@@ -1,5 +1,6 @@
 import torch
 
+
 class Event:
     r"""Wrapper around an MPS event.
 
@@ -16,7 +17,7 @@ def __init__(self, enable_timing=False):
 
     def __del__(self):
         # checks if torch._C is already destroyed
-        if hasattr(torch._C, '_mps_releaseEvent') and self.__eventId > 0:
+        if hasattr(torch._C, "_mps_releaseEvent") and self.__eventId > 0:
             torch._C._mps_releaseEvent(self.__eventId)
 
     def record(self):