fix: robustness to higher Python versions

Author: TianyiQ

src/abstractions/backends.py

@@ -1,17 +1,19 @@
 # Edit flashinfer cascade.py to make it compatible with Python 3.8
 from src.path import root
 import os
+import platform
 
-path = os.path.join(
-    os.environ["CONDA_PREFIX"], "lib/python3.8/site-packages/flashinfer/cascade.py"
-)
-with open(path, "r") as f:
-    content = f.read()
-content = content.replace("list[", "List[").replace(
-    "import Optional", "import List, Optional"
-)
-with open(path, "w") as f:
-    f.write(content)
+if "3.8" in platform.python_version():
+    path = os.path.join(
+        os.environ["CONDA_PREFIX"], "lib/python3.8/site-packages/flashinfer/cascade.py"
+    )
+    with open(path, "r") as f:
+        content = f.read()
+    content = content.replace("list[", "List[").replace(
+        "import Optional", "import List, Optional"
+    )
+    with open(path, "w") as f:
+        f.write(content)
 
 import torch
 import time