Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-09 23:12:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix test_debug_print

Browse source 
Follow-up of #173
This commit is contained in:
Colle 2024-01-12 18:32:22 +01:00 committed by GitHub
parent 457c3f5cbd
commit dba9065229
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/fluxion/layers/test_chain.py
View file

@ -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"
)