Skip to content

Commit 76efa8a

Browse files
committed
fix: fix HyperparameterTuner input strategy argument comparison
1 parent 09aa2e1 commit 76efa8a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/tuner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -741,7 +741,7 @@ def __init__(
741741
# For all other strategies for the backward compatibility we keep
742742
# the default value as 1 (previous default value).
743743
self.max_jobs = max_jobs
744-
if max_jobs is None and strategy is not GRID_SEARCH:
744+
if max_jobs is None and strategy != GRID_SEARCH:
745745
self.max_jobs = 1
746746
self.max_parallel_jobs = max_parallel_jobs
747747
self.max_runtime_in_seconds = max_runtime_in_seconds

0 commit comments

Comments
 (0)