From e6b2fbc40c889d9cff1cca10f846d83f2773bc9e Mon Sep 17 00:00:00 2001 From: huangzhenhua111 <1823301126@qq.com> Date: Tue, 14 Apr 2026 20:18:26 +0800 Subject: [PATCH] Fix backend loading logic in load_backend.py --- tensorlayerx/backend/ops/load_backend.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tensorlayerx/backend/ops/load_backend.py b/tensorlayerx/backend/ops/load_backend.py index 3fd2d0c..28cff92 100644 --- a/tensorlayerx/backend/ops/load_backend.py +++ b/tensorlayerx/backend/ops/load_backend.py @@ -32,10 +32,10 @@ path = os.path.join(tl_dir, 'tl_backend.json') with open(path, 'r') as load_f: load_dict = json.load(load_f) - if load_dict['backend'] is not config['backend']: - BACKEND = config['backend'] - else: + if load_dict.get('backend') != config['backend']: BACKEND = load_dict['backend'] + else: + BACKEND = config['backend'] # Set backend based on TL_BACKEND. if 'TL_BACKEND' in os.environ: