mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-10 07:21:59 +00:00
parent
457c3f5cbd
commit
dba9065229
|
@ -230,8 +230,8 @@ def test_setattr_dont_register() -> None:
|
||||||
|
|
||||||
|
|
||||||
EXPECTED_TREE = (
|
EXPECTED_TREE = (
|
||||||
"(CHAIN)\n ├── Linear(in_features=1, out_features=1) (x2)\n └── (CHAIN)\n ├── Linear(in_features=1,"
|
"(CHAIN)\n ├── Linear(in_features=1, out_features=1, device=cpu, dtype=float32) (x2)\n └── (CHAIN)\n ├── Linear(in_features=1,"
|
||||||
" out_features=1) #1\n └── Linear(in_features=2, out_features=1) #2"
|
" out_features=1, device=cpu, dtype=float32) #1\n └── Linear(in_features=2, out_features=1, device=cpu, dtype=float32) #2"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue