🎨 Format Python code with psf/black

This commit is contained in:
ndem0
2024-04-04 12:42:10 +00:00
committed by Nicola Demo
parent 5c50906771
commit 50fb3b731c

View File

@@ -56,11 +56,12 @@ class R3Refinement(Callback):
elif precision == "32-true":
precision = torch.float32
else:
raise RuntimeError("Currently R3Refinement is only implemented "
"for precision '32-true' and '64-true', set "
"Trainer precision to match one of the "
"available precisions.")
raise RuntimeError(
"Currently R3Refinement is only implemented "
"for precision '32-true' and '64-true', set "
"Trainer precision to match one of the "
"available precisions."
)
# compute residual
res_loss = {}