From 88bd79c33ec5dd321b85fead48e403df7f0c0744 Mon Sep 17 00:00:00 2001 From: nstarman Date: Fri, 22 Dec 2023 15:36:26 -0800 Subject: [PATCH] fix ruff Signed-off-by: nstarman --- pyproject.toml | 1 + src/jax_quantity/_register_primitives.py | 4 +--- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 2c7f51f..e9e4cc1 100644 --- 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 index cf7f519..f1028f6 100644 --- 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) )