diff --git a/tests/test_simple_math.py b/tests/test_simple_math.py index e575dad..1b14036 100644 --- a/tests/test_simple_math.py +++ b/tests/test_simple_math.py @@ -29,4 +29,4 @@ def closure(): optimizer.step(closure) pbar.set_postfix(mse_loss=loss.item(), reg_loss=reg_loss.item()) for layer in kan.layers: - print(layer.spline_weight) \ No newline at end of file + print(layer.spline_weight)