From cdd03db7b307f95a529e1c3fa84549fcd162baea Mon Sep 17 00:00:00 2001 From: arcticfly Date: Wed, 15 Oct 2025 16:52:05 -0700 Subject: [PATCH] Fix ruff formatting --- src/art/local/backend.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/art/local/backend.py b/src/art/local/backend.py index a470cbb7..13e80ec9 100644 --- a/src/art/local/backend.py +++ b/src/art/local/backend.py @@ -488,9 +488,9 @@ async def _train_model( num_gradient_steps = int( result.pop("num_gradient_steps", estimated_gradient_steps) ) - assert ( - num_gradient_steps == estimated_gradient_steps - ), f"num_gradient_steps {num_gradient_steps} != estimated_gradient_steps {estimated_gradient_steps}" + assert num_gradient_steps == estimated_gradient_steps, ( + f"num_gradient_steps {num_gradient_steps} != estimated_gradient_steps {estimated_gradient_steps}" + ) results.append(result) yield {**result, "num_gradient_steps": num_gradient_steps} pbar.update(1)