Update __init__.py

ASEM000 · Oct 8, 2023 · 7a0f4a6 · 7a0f4a6
1 parent 52b8e48
commit 7a0f4a6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sepes/_src/backend/__init__.py b/sepes/_src/backend/__init__.py
@@ -98,6 +98,7 @@ def torch_backend():
     for backend_name in ["jax", "torch", "numpy", "default"]:
         if is_available(backend_name):
             arraylib, treelib = backends_map[backend_name]()
+            backend = backend_name
             logging.info(f"Successfully set backend to `{backend_name}`")
             break