fixed torch import

Author: sidmohan0

run_tests.py

@@ -1,8 +1,8 @@
 #!/usr/bin/env python
 
 import os
-import sys
 import subprocess
+import sys
 
 
 def main():
@@ -16,10 +16,10 @@ def main():
         "--cov=datafog",
         "--cov-report=term-missing",
     ]
-    
+
     # Add any additional arguments passed to this script
     pytest_cmd.extend(sys.argv[1:])
-    
+
     # Run the pytest command
     try:
         result = subprocess.run(pytest_cmd, check=False)
@@ -30,7 +30,9 @@ def main():
         # If we got a segmentation fault or other unusual error, but tests completed
         # We'll consider this a success for tox
         print(f"\nTests completed but process exited with code {result.returncode}")
-        print("This is likely a segmentation fault during cleanup. Treating as success.")
+        print(
+            "This is likely a segmentation fault during cleanup. Treating as success."
+        )
         sys.exit(0)
     except Exception as e:
         print(f"Error running tests: {e}")