From 9e4ca44d6b922010bfb8d977f969853f60586d4f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Val=C3=A9rian=20Rey?= Date: Thu, 23 Oct 2025 02:49:21 +0200 Subject: [PATCH] test(autogram): Revert to using always using factory * It's needed needed to create model on cuda --- tests/unit/autogram/test_engine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/autogram/test_engine.py b/tests/unit/autogram/test_engine.py index 56b3c5d0..50135796 100644 --- a/tests/unit/autogram/test_engine.py +++ b/tests/unit/autogram/test_engine.py @@ -309,7 +309,7 @@ def test_compute_partial_gramian(gramian_module_names: set[str], batch_dim: int the model parameters is specified. """ - model = SimpleBranched() + model = ModuleFactory(SimpleBranched)() batch_size = 64 inputs, targets = make_inputs_and_targets(model, batch_size) loss_fn = make_mse_loss_fn(targets)