From 4c4d16f962ed350fff4e50aef77f5f3dd0599202 Mon Sep 17 00:00:00 2001 From: YutackPark Date: Tue, 30 Dec 2025 21:10:17 +0900 Subject: [PATCH] fix test sevennet --- tests/models/test_sevennet.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/models/test_sevennet.py b/tests/models/test_sevennet.py index c79419ff..1e373b79 100644 --- a/tests/models/test_sevennet.py +++ b/tests/models/test_sevennet.py @@ -34,8 +34,7 @@ def pretrained_sevenn_model(): """Load a pretrained SevenNet model for testing.""" cp = sevenn.util.load_checkpoint(model_name) - backend = "e3nn" - model_loaded = cp.build_model(backend) + model_loaded = cp.build_model() model_loaded.set_is_batch_data(True) return model_loaded.to(DEVICE)