Merge pull request #4275 from nzw0301/skip-running-trial-for-constraint

Skip constraint check for running trial
optuna · Jan 5, 2023 · 42493d9 · 42493d9
2 parents 91f21fb + 001c8c8
commit 42493d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optuna/samplers/_tpe/sampler.py b/optuna/samplers/_tpe/sampler.py
@@ -670,7 +670,7 @@ def _get_observation_pairs(
                 param_value = None
             values[param_name].append(param_value)
 
-        if constraints_enabled:
+        if constraints_enabled and trial.state != TrialState.RUNNING:
             assert violations is not None
             constraint = trial.system_attrs.get(_CONSTRAINTS_KEY)
             if constraint is None: