Skip to content

Commit d872030

Browse files
committed
feat: fix linting problems in the hyperparamter tuning job samples
1 parent 85622c9 commit d872030

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

samples/model-builder/create_hyperparameter_tuning_job_sample_test.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,11 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
from unittest.mock import ANY
16+
1517
import create_hyperparameter_tuning_job_sample
18+
1619
import test_constants as constants
17-
from unittest.mock import ANY
1820

1921

2022
def test_create_hyperparameter_tuning_job_sample(

samples/model-builder/test_constants.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -280,7 +280,7 @@
280280
# Hyperparameter tuning job
281281
HYPERPARAMETER_TUNING_JOB_DISPLAY_NAME = "hpt_job"
282282
HYPERPARAMETER_TUNING_JOB_ID = "4447046521673744384"
283-
HYPERPARAMETER_TUNING_JOB_METRIC_SPEC = {'loss': 'minimize',}
283+
HYPERPARAMETER_TUNING_JOB_METRIC_SPEC = {'loss': 'minimize'}
284284
HYPERPARAMETER_TUNING_JOB_MAX_TRIAL_COUNT = 128
285285
HYPERPARAMETER_TUNING_JOB_PARALLEL_TRIAL_COUNT = 8
286286
HYPERPARAMETER_TUNING_JOB_LABELS = {'my_key': 'my_value'}

0 commit comments

Comments
 (0)