Try AMD64 instead

cogment · Apr 19, 2024 · 0e65f61 · 0e65f61
1 parent 3483334
commit 0e65f61
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cogment_lab/cli/download_cogment.py b/cogment_lab/cli/download_cogment.py
@@ -31,10 +31,10 @@ class Arch(Enum):
 
 def get_current_arch() -> Arch:
     py_machine = platform.machine()
-    if py_machine in ["x86_64", "i686", "AMD64"]:
+    if py_machine in ["x86_64", "i686", "AMD64", "aarch64"]:
         return Arch.AMD64
 
-    if py_machine in ["arm64", "aarch64"]:
+    if py_machine in ["arm64"]:
         return Arch.ARM64
 
     raise RuntimeError(f"Unsupported architecture [{py_machine}]")