Fix linting

ratheron · ratheron · commit 86d1f84e6fa0 · 2025-12-12T12:55:48.000+01:00
diff --git a/tests/unit/test_gradients.py b/tests/unit/test_gradients.py
@@ -1,7 +1,5 @@
 from __future__ import annotations
 
-import warnings
-
 import jax
 import jax.numpy as jnp
 import pytest
