Skip to content

Commit aa21da9

Browse files
committed
refactor: improve assertion clarity in LocalBackend gradient step validation
- Simplified the assertion statement for validating num_gradient_steps against estimated_gradient_steps for better readability.
1 parent c2f0e39 commit aa21da9

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/art/local/backend.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -607,9 +607,9 @@ async def _train_model(
607607
num_gradient_steps = int(
608608
result.pop("num_gradient_steps", estimated_gradient_steps)
609609
)
610-
assert (
611-
num_gradient_steps == estimated_gradient_steps
612-
), f"num_gradient_steps {num_gradient_steps} != estimated_gradient_steps {estimated_gradient_steps}"
610+
assert num_gradient_steps == estimated_gradient_steps, (
611+
f"num_gradient_steps {num_gradient_steps} != estimated_gradient_steps {estimated_gradient_steps}"
612+
)
613613
results.append(result)
614614
yield {**result, "num_gradient_steps": num_gradient_steps}
615615
pbar.update(1)

0 commit comments

Comments
 (0)