From fd7f77abf1f9bc917d39372db60511c5e568cede Mon Sep 17 00:00:00 2001
From: Anda Zhou <83614683+azhou-determined@users.noreply.github.com>
Date: Thu, 12 Sep 2024 09:50:50 -0700
Subject: [PATCH] fix: move validation dataloader check in PyTorchTrial
 [MD-515] (#9923)

---
 harness/determined/pytorch/_pytorch_trial.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/harness/determined/pytorch/_pytorch_trial.py b/harness/determined/pytorch/_pytorch_trial.py
index 42512f1fd2e..1363f321483 100644
--- a/harness/determined/pytorch/_pytorch_trial.py
+++ b/harness/determined/pytorch/_pytorch_trial.py
@@ -992,8 +992,8 @@ def _validate(self, searcher_op: Optional[core.SearcherOperation] = None) -> Dic
 
         else:
             assert self._evaluate_full_dataset_defined(), "evaluate_full_dataset not defined."
-            assert self.validation_loader is not None
             if self.is_chief:
+                assert self.validation_loader is not None
                 metrics = self.trial.evaluate_full_dataset(data_loader=self.validation_loader)
 
                 if not isinstance(metrics, dict):