From 563aa94db3829faad4fbb2f3bcec822cb5750589 Mon Sep 17 00:00:00 2001 From: GiovanniCanali Date: Thu, 11 Dec 2025 13:16:44 +0100 Subject: [PATCH] fix checks --- pina/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pina/trainer.py b/pina/trainer.py index 8e1d95110..e92928d1e 100644 --- a/pina/trainer.py +++ b/pina/trainer.py @@ -330,7 +330,7 @@ def _check_consistency_and_set_defaults( else: pin_memory = False if num_workers is not None: - check_consistency(pin_memory, int) + check_consistency(num_workers, int) else: num_workers = 0 if shuffle is not None: