move Trains logger to Bolts (#2384)

* move Trains logger * chlog
Lightning-AI · Jun 27, 2020 · 41f5df1 · 41f5df1
1 parent 4e13e41
commit 41f5df1
Show file tree

Hide file tree

Showing 11 changed files with 3 additions and 510 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -16,6 +16,8 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 ### Removed
 
+- Moved `TrainsLogger` to Bolts ([#2384](https://github.com/PyTorchLightning/pytorch-lightning/pull/2384))
+
 ### Fixed
 
 - Fixed parsing TPU arguments and TPU tests ([#2094](https://github.com/PyTorchLightning/pytorch-lightning/pull/2094))

diff --git a/README.md b/README.md
@@ -329,7 +329,6 @@ Lightning has out-of-the-box integration with the popular logging/visualizing fr
 - [Neptune.ai](https://neptune.ai/)
 - [Comet.ml](https://www.comet.ml/site/)
 - [Wandb](https://www.wandb.com/)
-- [Trains](https://github.com/allegroai/trains)
 - ...  
 
 ![tensorboard-support](docs/source/_images/general/tf_loss.png)

diff --git a/docs/source/experiment_logging.rst b/docs/source/experiment_logging.rst
@@ -116,50 +116,6 @@ The :class:`~pytorch_lightning.loggers.NeptuneLogger` is available anywhere exce
 
 ----------------
 
-allegro.ai TRAINS
-^^^^^^^^^^^^^^^^^
-
-`allegro.ai <https://github.com/allegroai/trains/>`_ is a third-party logger.
-To use :class:`~pytorch_lightning.loggers.TrainsLogger` as your logger do the following.
-First, install the package:
-
-.. code-block:: bash
-
-    pip install trains
-
-Then configure the logger and pass it to the :class:`~pytorch_lightning.trainer.trainer.Trainer`:
-
-.. testcode::
-
-    from pytorch_lightning.loggers import TrainsLogger
-    trains_logger = TrainsLogger(
-        project_name='examples',
-        task_name='pytorch lightning test',
-    )
-    trainer = Trainer(logger=trains_logger)
-
-.. testoutput::
-    :options: +ELLIPSIS, +NORMALIZE_WHITESPACE
-    :hide:
-
-    TRAINS Task: ...
-    TRAINS results page: ...
-
-The :class:`~pytorch_lightning.loggers.TrainsLogger` is available anywhere in your
-:class:`~pytorch_lightning.core.lightning.LightningModule`.
-
-.. testcode::
-
-    class MyModule(LightningModule):
-        def __init__(self):
-            some_img = fake_image()
-            self.logger.experiment.log_image('debug', 'generated_image_0', some_img, 0)
-
-.. seealso::
-    :class:`~pytorch_lightning.loggers.TrainsLogger` docs.
-
-----------------
-
 Tensorboard
 ^^^^^^^^^^^
 

diff --git a/docs/source/experiment_reporting.rst b/docs/source/experiment_reporting.rst
@@ -43,7 +43,7 @@ want to log using this trainer flag.
 Log metrics
 ^^^^^^^^^^^
 
-To plot metrics into whatever logger you passed in (tensorboard, comet, neptune, TRAINS, etc...)
+To plot metrics into whatever logger you passed in (tensorboard, comet, neptune, etc...)
 
 1. training_epoch_end, validation_epoch_end, test_epoch_end will all log anything in the "log" key of the return dict.
 

diff --git a/docs/source/loggers.rst b/docs/source/loggers.rst
@@ -138,10 +138,4 @@ Test-tube
 ^^^^^^^^^
 
 .. autoclass:: pytorch_lightning.loggers.test_tube.TestTubeLogger
-    :noindex:
-
-Trains
-^^^^^^
-
-.. autoclass:: pytorch_lightning.loggers.trains.TrainsLogger
     :noindex:
diff --git a/environment.yml b/environment.yml
@@ -37,4 +37,3 @@ dependencies:
         - comet_ml>=1.0.56
         - wandb>=0.8.21
         - neptune-client>=0.4.4
-        - trains>=0.13.3
diff --git a/pytorch_lightning/loggers/__init__.py b/pytorch_lightning/loggers/__init__.py
@@ -46,10 +46,3 @@
     pass  # pragma: no-cover
 else:
     __all__.append('WandbLogger')
-
-try:
-    from pytorch_lightning.loggers.trains import TrainsLogger
-except ImportError:  # pragma: no-cover
-    pass  # pragma: no-cover
-else:
-    __all__.append('TrainsLogger')