wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

prune tests (#564)

Browse Source 
* format docstring in tests

* prune unused vars

* optimize imports

* drop duplicated var
This commit is contained in:
Jirka Borovec
2019-12-04 06:48:53 -05:00
committed by William Falcon
parent 62f6f92fdf
commit 63717e8fda
8 changed files with 108 additions and 292 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_logging.py
+10 -29
View File
@@ -1,26 +1,19 @@
import os
import pickle
import numpy as np
import torch
from pytorch_lightning import Trainer
from pytorch_lightning.testing import LightningTestModel
from pytorch_lightning.logging import LightningLoggerBase, rank_zero_only
import tests.utils as tutils
from pytorch_lightning import Trainer
from pytorch_lightning.logging import LightningLoggerBase, rank_zero_only
from pytorch_lightning.testing import LightningTestModel
def test_testtube_logger(tmpdir):
"""
verify that basic functionality of test tube logger works
"""
"""Verify that basic functionality of test tube logger works."""
tutils.reset_seed()
hparams = tutils.get_hparams()
model = LightningTestModel(hparams)
save_dir = tmpdir
logger = tutils.get_test_tube_logger(save_dir, False)
logger = tutils.get_test_tube_logger(tmpdir, False)
trainer_options = dict(
max_nb_epochs=1,
@@ -35,16 +28,12 @@ def test_testtube_logger(tmpdir):
def test_testtube_pickle(tmpdir):
"""
Verify that pickling a trainer containing a test tube logger works
"""
"""Verify that pickling a trainer containing a test tube logger works."""
tutils.reset_seed()
hparams = tutils.get_hparams()
model = LightningTestModel(hparams)
save_dir = tmpdir
logger = tutils.get_test_tube_logger(tmpdir, False)
logger.log_hyperparams(hparams)
logger.save()
@@ -62,9 +51,7 @@ def test_testtube_pickle(tmpdir):
def test_mlflow_logger(tmpdir):
"""
verify that basic functionality of mlflow logger works
"""
"""Verify that basic functionality of mlflow logger works."""
tutils.reset_seed()
try:
@@ -93,9 +80,7 @@ def test_mlflow_logger(tmpdir):
def test_mlflow_pickle(tmpdir):
"""
verify that pickling trainer with mlflow logger works
"""
"""Verify that pickling trainer with mlflow logger works."""
tutils.reset_seed()
try:
@@ -122,9 +107,7 @@ def test_mlflow_pickle(tmpdir):
def test_comet_logger(tmpdir):
"""
verify that basic functionality of Comet.ml logger works
"""
"""Verify that basic functionality of Comet.ml logger works."""
tutils.reset_seed()
try:
@@ -158,9 +141,7 @@ def test_comet_logger(tmpdir):
def test_comet_pickle(tmpdir):
"""
verify that pickling trainer with comet logger works
"""
"""Verify that pickling trainer with comet logger works."""
tutils.reset_seed()
try: