use_existing_torch.py: filter out comments

tdoublep · tdoublep · commit 5d914f68f566 · 2025-01-21T10:05:22.000+01:00
Signed-off-by: Thomas Parnell &lt;tpa@zurich.ibm.com&gt;
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:
