From dba90652298db8dc39ae84682d4d335a55aa0842 Mon Sep 17 00:00:00 2001 From: Colle Date: Fri, 12 Jan 2024 18:32:22 +0100 Subject: [PATCH] fix test_debug_print Follow-up of #173 --- tests/fluxion/layers/test_chain.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/fluxion/layers/test_chain.py b/tests/fluxion/layers/test_chain.py index 5b5cd1f..402564c 100644 --- a/tests/fluxion/layers/test_chain.py +++ b/tests/fluxion/layers/test_chain.py @@ -230,8 +230,8 @@ def test_setattr_dont_register() -> None: EXPECTED_TREE = ( - "(CHAIN)\n ├── Linear(in_features=1, out_features=1) (x2)\n └── (CHAIN)\n ├── Linear(in_features=1," - " out_features=1) #1\n └── Linear(in_features=2, out_features=1) #2" + "(CHAIN)\n ├── Linear(in_features=1, out_features=1, device=cpu, dtype=float32) (x2)\n └── (CHAIN)\n ├── Linear(in_features=1," + " out_features=1, device=cpu, dtype=float32) #1\n └── Linear(in_features=2, out_features=1, device=cpu, dtype=float32) #2" )