2020-03-03 01:49:14 +00:00
|
|
|
from pytorch_lightning.loggers import MLFlowLogger
|
|
|
|
|
|
|
|
|
|
|
2020-04-15 00:32:33 +00:00
|
|
|
def test_mlflow_logger_exists(tmpdir):
|
2020-03-03 01:49:14 +00:00
|
|
|
"""Verify that basic functionality of mlflow logger works."""
|
2020-04-15 00:32:33 +00:00
|
|
|
logger = MLFlowLogger('test', save_dir=tmpdir)
|
2020-03-03 01:49:14 +00:00
|
|
|
# Test already exists
|
2020-04-15 00:32:33 +00:00
|
|
|
logger2 = MLFlowLogger('test', save_dir=tmpdir)
|
|
|
|
|
assert logger.run_id != logger2.run_id
|
