From 7459c092187b5e90f30a8757f8a2a4a915a40b04 Mon Sep 17 00:00:00 2001 From: Doug Blank Date: Tue, 11 Feb 2020 13:35:22 -0500 Subject: [PATCH] Use standard Comet env variable names in docstring (#816) * COMET_KEY -> COMET_API_KEY * COMET_REST_KEY -> COMET_REST_API_KEY --- pytorch_lightning/loggers/comet.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/loggers/comet.py b/pytorch_lightning/loggers/comet.py index b3d59eb1fe..ce547ab66f 100644 --- a/pytorch_lightning/loggers/comet.py +++ b/pytorch_lightning/loggers/comet.py @@ -43,10 +43,10 @@ class CometLogger(LightningLoggerBase): from pytorch_lightning.loggers import CometLogger # arguments made to CometLogger are passed on to the comet_ml.Experiment class comet_logger = CometLogger( - api_key=os.environ["COMET_KEY"], + api_key=os.environ["COMET_API_KEY"], workspace=os.environ["COMET_WORKSPACE"], # Optional project_name="default_project", # Optional - rest_api_key=os.environ["COMET_REST_KEY"], # Optional + rest_api_key=os.environ["COMET_REST_API_KEY"], # Optional experiment_name="default" # Optional ) trainer = Trainer(logger=comet_logger) @@ -60,7 +60,7 @@ class CometLogger(LightningLoggerBase): save_dir=".", workspace=os.environ["COMET_WORKSPACE"], # Optional project_name="default_project", # Optional - rest_api_key=os.environ["COMET_REST_KEY"], # Optional + rest_api_key=os.environ["COMET_REST_API_KEY"], # Optional experiment_name="default" # Optional ) trainer = Trainer(logger=comet_logger)