use_existing_torch.py: filter out comments

Signed-off-by: Thomas Parnell <[email protected]>
vllm-project · Jan 21, 2025 · 3105f0c · 3105f0c
1 parent f2e9f2a
commit 3105f0c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/use_existing_torch.py b/use_existing_torch.py
@@ -6,6 +6,8 @@
     print(f">>> cleaning {file}")
     with open(file) as f:
         lines = f.readlines()
+    # filter out comments
+    lines = [ line.split(" #")[0] for line in lines ]
     if "torch" in "".join(lines).lower():
         print("removed:")
         with open(file, 'w') as f: