def test_set_data_dir(self):
path = "./dataset"
environment.set_data_dir(path)
abs_path = os.path.abspath(path)
assert environment.DATA_DIR == abs_path
assert os.environ["DATA_DIR"] == abs_path
def train(config, experiment_id, recreate, profile_step, data_dir, output_dir):
environment.set_data_dir(data_dir)
environment.set_output_dir(output_dir)
experiment_id, checkpoint_name = run_train(config, experiment_id, recreate,
profile_step)
click.echo('Next step: blueoil convert -e {} -p {}'.format(
experiment_id, checkpoint_name))
def test_set_data_dir_with_gcs_path(self):
path = "gs://dataset"
environment.set_data_dir(path)
assert environment.DATA_DIR == path
assert os.environ["DATA_DIR"] == path