From d1633aac112fb35878aae83deb8b57259acea75b Mon Sep 17 00:00:00 2001 From: Jay Morgan Date: Wed, 11 Dec 2019 03:24:21 +0300 Subject: [PATCH] Fix #618 Change papi to api (#619) * Change papi to api * Added try catch for old/new api reference --- pytorch_lightning/logging/comet.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/logging/comet.py b/pytorch_lightning/logging/comet.py index b7476066..3fe25418 100644 --- a/pytorch_lightning/logging/comet.py +++ b/pytorch_lightning/logging/comet.py @@ -52,7 +52,11 @@ from logging import getLogger try: from comet_ml import Experiment as CometExperiment from comet_ml import OfflineExperiment as CometOfflineExperiment - from comet_ml.papi import API + try: + from comet_ml.api import API + except ImportError: + # For more information, see: https://www.comet.ml/docs/python-sdk/releases/#release-300 + from comet_ml.papi import API except ImportError: raise ImportError('Missing comet_ml package.')