diff --git a/tests/unit/test_cuda_graph_readiness.py b/tests/unit/test_cuda_graph_readiness.py index 84e265fa..c8ab5726 100644 --- a/tests/unit/test_cuda_graph_readiness.py +++ b/tests/unit/test_cuda_graph_readiness.py @@ -147,7 +147,7 @@ def run_sync_debug_mode(): ("ffn", ffn_mhcs, f"model.layers.{li}.ffn_hc.fn", f"model.layers.{li}.ffn_hc.base", f"model.layers.{li}.ffn_hc.scale"), ]: fn, base, scale = all_w.get(fn_s), all_w.get(base_s), all_w.get(scale_s) - if fn is not None and base is not None and scale is not not None: + if fn is not None and base is not None and scale is not None: m = mHCLayer(hidden_dim=H, n_hc=4, t_max_sinkhorn=20, device=dev) n = 4 m.load_weights(