Cleaned up some ruff issues.

plasma-umass · Nov 4, 2024 · f28e6ac · f28e6ac
1 parent 024e8ff
commit f28e6ac
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/scalene/scalene_leak_analysis.py b/scalene/scalene_leak_analysis.py
@@ -1,4 +1,4 @@
-from typing import Any, List, OrderedDict, Tuple
+from typing import List, OrderedDict, Tuple
 
 from scalene.scalene_statistics import Filename, LineNumber, ScaleneStatistics
 

diff --git a/scalene/scalene_neuron.py b/scalene/scalene_neuron.py
@@ -103,8 +103,7 @@ def has_gpu(self) -> bool:
                 return False
             else:
                 return True
-        except subprocess.CalledProcessError as e:
-            # print(f"Error running neuron-ls: {e}")
+        except subprocess.CalledProcessError:
             return False
         except FileNotFoundError:
             # print("neuron-ls command not found. Ensure AWS Neuron SDK is installed.")

diff --git a/scalene/scalene_statistics.py b/scalene/scalene_statistics.py
@@ -7,7 +7,6 @@
 from collections import defaultdict
 from typing import (
     Any,
-    DefaultDict,
     Dict,
     List,
     NewType,