Skip to content

Commit 8ee2ed4

Browse files
committed
Fix import in forward_backwards.py
1 parent 96ece7a commit 8ee2ed4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/utils/forward_backwards.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from torch.utils._pytree import PyTree, tree_flatten, tree_map
77

88
from torchjd.aggregation import Aggregator, Weighting
9-
from torchjd.autogram._engine import Engine
9+
from torchjd.autogram import Engine
1010
from torchjd.autojac import backward
1111

1212

0 commit comments

Comments
 (0)