def create(file_store_path, experiment_name):
"""
Creates a new experiment in FileStore backend.
"""
fs = file_store.FileStore(file_store_path)
exp_id = fs.create_experiment(experiment_name)
print("Created experiment '%s' with id '%d'" % (experiment_name, exp_id))
def create(file_store, artifact_root, experiment_name):
"""
Creates a new experiment in FileStore backend.
"""
fs = store.FileStore(file_store, artifact_root)
exp_id = fs.create_experiment(experiment_name)
print("Created experiment '%s' with id %d" % (experiment_name, exp_id))
def list_experiments(file_store):
"""
List all experiment in FileStore backend.
"""
fs = store.FileStore(file_store)
experiments = fs.list_experiments()
table = [[exp.experiment_id, exp.name, os.path.abspath(exp.artifact_location)]
for exp in experiments]
print(tabulate(sorted(table), headers=["Experiment Id", "Name", "Artifact Location"]))
def test_docker_project_tracking_uri_propagation(
ProfileConfigProvider, tmpdir, tracking_uri,
expected_command_segment): # pylint: disable=unused-argument
build_docker_example_base_image()
mock_provider = mock.MagicMock()
mock_provider.get_config.return_value = \
DatabricksConfig("host", "user", "pass", None, insecure=True)
ProfileConfigProvider.return_value = mock_provider
# Create and mock local tracking directory
local_tracking_dir = os.path.join(tmpdir.strpath, "mlruns")
if tracking_uri is None:
tracking_uri = local_tracking_dir
old_uri = mlflow.get_tracking_uri()
try:
mlflow.set_tracking_uri(tracking_uri)
with mock.patch("mlflow.tracking.utils._get_store") as _get_store_mock:
_get_store_mock.return_value = file_store.FileStore(local_tracking_dir)
mlflow.projects.run(TEST_DOCKER_PROJECT_DIR, experiment_id=0)
finally:
mlflow.set_tracking_uri(old_uri)