__init__.py

# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

import os

os.environ['NNI_PLATFORM'] = 'unittest'
os.environ['NNI_TRIAL_JOB_ID'] = 'test_trial_job_id'