fix ruff

Signed-off-by: nstarman <nstarman@users.noreply.github.com>
GalacticDynamics · Dec 22, 2023 · 88bd79c · 88bd79c
1 parent f0a22b0
commit 88bd79c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -131,6 +131,7 @@ ignore = [
   "ANN101",   # Missing type annotation for `self` in method
   "ANN401",   # Dynamically typed expressions (typing.Any) are disallowed
   "ARG001",   # Unused function argument  # TODO: resolve
+  "COM812",   # <- for ruff.format
   "D103",     # Missing docstring in public function  # TODO: resolve
   "D105",     # Missing docstring in magic method
   "D203",     # 1 blank line required before class docstring

diff --git a/src/jax_quantity/_register_primitives.py b/src/jax_quantity/_register_primitives.py
@@ -32,9 +32,7 @@ def register(primitive: jax.core.Primitive) -> Callable[[T], T]:
 
 def _to_value_rad_or_one(q: Quantity) -> ArrayLike:
     return (
-        q.to_value(radian)
-        if can_convert(q.unit, radian)
-        else q.to_value(dimensionless)
+        q.to_value(radian) if can_convert(q.unit, radian) else q.to_value(dimensionless)
     )