From 9e7c6a0465607450c3821d473f7306bf35887c89 Mon Sep 17 00:00:00 2001 From: chaithyagr Date: Tue, 20 Apr 2021 09:53:53 +0200 Subject: [PATCH] Minor bug fix, remove elif --- modopt/base/backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modopt/base/backend.py b/modopt/base/backend.py index def77151..5fbe912f 100644 --- a/modopt/base/backend.py +++ b/modopt/base/backend.py @@ -95,7 +95,7 @@ def get_array_module(input_data): if LIBRARIES['tensorflow'] is not None: if isinstance(input_data, LIBRARIES['tensorflow'].ndarray): return LIBRARIES['tensorflow'] - elif LIBRARIES['cupy'] is not None: + if LIBRARIES['cupy'] is not None: if isinstance(input_data, LIBRARIES['cupy'].ndarray): return LIBRARIES['cupy'] return np