From 8c16e27ae468b678b394a0eff11e066c5e341b12 Mon Sep 17 00:00:00 2001 From: Nicola Demo Date: Wed, 2 Aug 2023 10:04:02 +0200 Subject: [PATCH] Update test_pinn.py --- tests/test_solvers/test_pinn.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/test_solvers/test_pinn.py b/tests/test_solvers/test_pinn.py index c8dcee2..3b8d0a0 100644 --- a/tests/test_solvers/test_pinn.py +++ b/tests/test_solvers/test_pinn.py @@ -104,7 +104,6 @@ def test_train_restore(): pinn = PINN(problem = poisson_problem, model=model, extra_features=None, loss=LpLoss()) trainer = Trainer(solver=pinn, max_epochs=5, accelerator='cpu', default_root_dir=tmpdir) trainer.train() - print('ggg') ntrainer = Trainer(solver=pinn, max_epochs=15, accelerator='cpu') t = ntrainer.train( ckpt_path=f'{tmpdir}/lightning_logs/version_0/checkpoints/epoch=4-step=5.ckpt')