From 10e4b18452e9822a06baabdc727ce56ab521cac8 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 7 Aug 2019 10:14:59 -0400 Subject: [PATCH] made imports absolute --- pytorch_lightning/callbacks/pt_callbacks.py | 2 +- pytorch_lightning/models/trainer.py | 8 ++++---- pytorch_lightning/root_module/model_saving.py | 2 +- pytorch_lightning/root_module/root_module.py | 10 +++++----- pytorch_lightning/testing/lm_test_module.py | 2 +- 5 files changed, 12 insertions(+), 12 deletions(-) diff --git a/pytorch_lightning/callbacks/pt_callbacks.py b/pytorch_lightning/callbacks/pt_callbacks.py index f07c6a28..24d46e75 100644 --- a/pytorch_lightning/callbacks/pt_callbacks.py +++ b/pytorch_lightning/callbacks/pt_callbacks.py @@ -3,7 +3,7 @@ import shutil import numpy as np -from ..pt_overrides.override_data_parallel import LightningDistributedDataParallel +from pytorch_lightning.pt_overrides.override_data_parallel import LightningDistributedDataParallel class Callback(object): diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 1309e4c5..1b3a9a7a 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -13,11 +13,11 @@ from torch.utils.data.distributed import DistributedSampler import torch.multiprocessing as mp import torch.distributed as dist -from ..root_module.memory import get_gpu_memory_map -from ..root_module.model_saving import TrainerIO -from ..pt_overrides.override_data_parallel import ( +from pytorch_lightning.root_module.memory import get_gpu_memory_map +from pytorch_lightning.root_module.model_saving import TrainerIO +from pytorch_lightning.pt_overrides.override_data_parallel import ( LightningDistributedDataParallel, LightningDataParallel) -from ..utilities.debugging import MisconfigurationException +from pytorch_lightning.utilities.debugging import MisconfigurationException try: from apex import amp diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index ffd76387..7d6d2c8c 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -3,7 +3,7 @@ import re import torch -from ..pt_overrides.override_data_parallel import ( +from pytorch_lightning.pt_overrides.override_data_parallel import ( LightningDistributedDataParallel, LightningDataParallel) diff --git a/pytorch_lightning/root_module/root_module.py b/pytorch_lightning/root_module/root_module.py index 96dbfdcd..700e6db1 100644 --- a/pytorch_lightning/root_module/root_module.py +++ b/pytorch_lightning/root_module/root_module.py @@ -1,10 +1,10 @@ import torch -from .memory import ModelSummary -from .grads import GradInformation -from .model_saving import ModelIO, load_hparams_from_tags_csv -from .hooks import ModelHooks -from .decorators import data_loader +from pytorch_lightning.root_module.memory import ModelSummary +from pytorch_lightning.root_module.grads import GradInformation +from pytorch_lightning.root_module.model_saving import ModelIO, load_hparams_from_tags_csv +from pytorch_lightning.root_module.hooks import ModelHooks +from pytorch_lightning.root_module.decorators import data_loader class LightningModule(GradInformation, ModelIO, ModelHooks): diff --git a/pytorch_lightning/testing/lm_test_module.py b/pytorch_lightning/testing/lm_test_module.py index 61ecf874..8fe4cfc0 100644 --- a/pytorch_lightning/testing/lm_test_module.py +++ b/pytorch_lightning/testing/lm_test_module.py @@ -11,7 +11,7 @@ from torchvision.datasets import MNIST from torchvision import transforms from test_tube import HyperOptArgumentParser -from ..root_module.root_module import LightningModule +from pytorch_lightning.root_module.root_module import LightningModule from pytorch_lightning import data_loader