Skip to content

fix: fix optree compatibility

Codecov / codecov/patch succeeded Nov 8, 2023 in 1s

95.45% of diff hit (target 100.00%)

View this Pull Request on Codecov

95.45% of diff hit (target 100.00%)

Annotations

Check warning on line 70 in torchopt/nn/stateless.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

torchopt/nn/stateless.py#L69-L70

Added lines #L69 - L70 were not covered by tests

Check warning on line 160 in torchopt/transform/trace.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

torchopt/transform/trace.py#L160

Added line #L160 was not covered by tests

Check warning on line 177 in torchopt/transform/trace.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

torchopt/transform/trace.py#L177

Added line #L177 was not covered by tests

Check warning on line 195 in torchopt/transform/trace.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

torchopt/transform/trace.py#L195

Added line #L195 was not covered by tests