Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use standard Comet env variable names in docstring (#816) 

* COMET_KEY -> COMET_API_KEY
* COMET_REST_KEY -> COMET_REST_API_KEY
This commit is contained in:
Doug Blank 2020-02-11 13:35:22 -05:00 committed by GitHub
parent 64d1bac7d6
commit 7459c09218
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pytorch_lightning/loggers/comet.py
View File

@ -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)