Lint test galore (#1507)

pytorch · Jan 7, 2025 · 416cdb6 · 416cdb6
1 parent 6399320
commit 416cdb6
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/ruff.toml b/ruff.toml
@@ -21,6 +21,7 @@ include = [
     "test/sparsity/**/*.py",
     "test/prototype/**/*.py",
     "test/dora/**/*.py",
+    "test/galore/**/*.py",
 ]
 
 lint.select = ["F", "I"]

diff --git a/test/galore/profile_memory_usage.py b/test/galore/profile_memory_usage.py
@@ -163,7 +163,7 @@ def run(args, file_prefix):
     if args.torch_profiler:
         print(f"Finished profiling, outputs saved to {args.output_dir}/{file_prefix}*")
     else:
-        print(f"Finished profiling")
+        print("Finished profiling")
 
 
 if __name__ == "__main__":

diff --git a/test/galore/profiling_utils.py b/test/galore/profiling_utils.py
@@ -75,7 +75,6 @@ def get_cuda_memory_usage(units="MB", show=True):
 
 
 def export_memory_snapshot(prefix) -> None:
-
     # Prefix for file names.
     timestamp = datetime.now().strftime(TIME_FORMAT_STR)
     file_prefix = f"{prefix}_{timestamp}"
@@ -115,7 +114,6 @@ def trace_handler(
     export_memory_timeline=True,
     print_table=True,
 ):
-
     timestamp = datetime.now().strftime(TIME_FORMAT_STR)
     file_prefix = os.path.join(output_dir, f"{prefix}_{timestamp}")