We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 153d7e8 commit 64e4731Copy full SHA for 64e4731
src/art/local/backend.py
@@ -488,9 +488,9 @@ async def _train_model(
488
num_gradient_steps = int(
489
result.pop("num_gradient_steps", estimated_gradient_steps)
490
)
491
- assert (
492
- num_gradient_steps == estimated_gradient_steps
493
- ), 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}"
+ )
494
results.append(result)
495
yield {**result, "num_gradient_steps": num_gradient_steps}
496
pbar.update(1)
0 commit comments