Fix the get_device() function for the torch backend (keras-team#418)

freedomtan · Jun 30, 2023 · 87b97e5 · 87b97e5
1 parent 638a4fb
commit 87b97e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_core/backend/torch/core.py b/keras_core/backend/torch/core.py
@@ -46,7 +46,7 @@ def get_default_device():
 def get_device():
     device = global_state.get_global_attribute("torch_device", None)
     if device is None:
-        get_default_device()
+        return get_default_device()
     return device