fix imports

Author: Borda

pytorch_lightning/core/lightning.py

@@ -20,7 +20,7 @@
 from pytorch_lightning.core.memory import ModelSummary
 from pytorch_lightning.core.saving import ModelIO, load_hparams_from_tags_csv
 from pytorch_lightning.overrides.data_parallel import LightningDistributedDataParallel
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 
 try:
     import torch_xla.core.xla_model as xm

pytorch_lightning/debugging/__init__.py

@@ -113,9 +113,11 @@ def custom_processing_step(self, data):
 
 """
 
+from pytorch_lightning.debugging.exceptions import MisconfigurationException
 from pytorch_lightning.debugging.profilers import BaseProfiler, Profiler, AdvancedProfiler, PassThroughProfiler
 
 __all__ = [
+    'MisconfigurationException',
     'BaseProfiler',
     'Profiler',
     'AdvancedProfiler',

pytorch_lightning/debugging/profilers.py

@@ -166,12 +166,12 @@ class AdvancedProfiler(BaseProfiler):
     verbose and you should only use this if you want very detailed reports.
     """
 
-    def __init__(self, output_filename=None, line_count_restriction=1.0):
+    def __init__(self, output_filename: str = None, line_count_restriction: float = 1.0):
         """
         Prams:
-            output_filename (str): optionally save profile results to file instead of printing
+            output_filename: optionally save profile results to file instead of printing
                 to std out when training is finished.
-            line_count_restriction (int|float): this can be used to limit the number of functions
+            line_count_restriction: this can be used to limit the number of functions
                 reported for each action. either an integer (to select a count of lines),
                 or a decimal fraction between 0.0 and 1.0 inclusive (to select a percentage of lines)
         """
@@ -200,16 +200,16 @@ def stop(self, action_name: str) -> None:
         pr.disable()
 
     def summary(self) -> str:
-        self.recorded_stats = {}
+        recorded_stats = {}
         for action_name, pr in self.profiled_actions.items():
             s = io.StringIO()
             ps = pstats.Stats(pr, stream=s).strip_dirs().sort_stats('cumulative')
             ps.print_stats(self.line_count_restriction)
-            self.recorded_stats[action_name] = s.getvalue()
+            recorded_stats[action_name] = s.getvalue()
 
         # log to standard out
         output_string = f"{os.linesep}Profiler Report{os.linesep}"
-        for action, stats in self.recorded_stats.items():
+        for action, stats in recorded_stats.items():
             output_string += f"{os.linesep}Profile stats for: {action}{os.linesep}{stats}"
 
         return output_string

pytorch_lightning/loggers/comet.py

@@ -28,7 +28,7 @@
 
 from pytorch_lightning import _logger as log
 from pytorch_lightning.loggers.base import LightningLoggerBase, rank_zero_only
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 
 
 class CometLogger(LightningLoggerBase):

pytorch_lightning/trainer/data_loading.py

@@ -6,7 +6,7 @@
 from torch.utils.data.distributed import DistributedSampler
 
 from pytorch_lightning.core import LightningModule
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 
 try:
     from apex import amp

pytorch_lightning/trainer/distrib_data_parallel.py

@@ -122,7 +122,7 @@ def train_fx(trial_hparams, cluster_manager, _):
 import torch
 from pytorch_lightning import _logger as log
 from pytorch_lightning.loggers import LightningLoggerBase
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 
 try:
     from apex import amp

pytorch_lightning/trainer/distrib_parts.py

@@ -344,7 +344,7 @@
     LightningDistributedDataParallel,
     LightningDataParallel,
 )
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 
 try:
     from apex import amp

pytorch_lightning/trainer/evaluation_loop.py

@@ -135,7 +135,7 @@
 
 from pytorch_lightning.core.lightning import LightningModule
 from pytorch_lightning.overrides.data_parallel import LightningDistributedDataParallel, LightningDataParallel
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 
 try:
     import torch_xla.distributed.parallel_loader as xla_pl

pytorch_lightning/trainer/trainer.py

@@ -32,7 +32,7 @@
 from pytorch_lightning.trainer.training_io import TrainerIOMixin
 from pytorch_lightning.trainer.training_loop import TrainerTrainLoopMixin
 from pytorch_lightning.trainer.training_tricks import TrainerTrainingTricksMixin
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 
 try:
     from apex import amp

pytorch_lightning/trainer/training_loop.py

@@ -145,7 +145,7 @@ def training_step(self, batch, batch_idx):
 from pytorch_lightning.callbacks.base import Callback
 from pytorch_lightning.core.lightning import LightningModule
 from pytorch_lightning.loggers import LightningLoggerBase
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 
 try:
     from apex import amp

tests/loggers/test_comet.py

@@ -8,7 +8,7 @@
 import tests.base.utils as tutils
 from pytorch_lightning import Trainer
 from pytorch_lightning.loggers import CometLogger
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 from tests.base import LightningTestModel
 
 

tests/models/test_amp.py

@@ -4,7 +4,7 @@
 
 import tests.base.utils as tutils
 from pytorch_lightning import Trainer
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 from tests.base import (
     LightningTestModel,
 )

tests/models/test_gpu.py

@@ -11,7 +11,7 @@
     parse_gpu_ids,
     determine_root_gpu_device,
 )
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 from tests.base import LightningTestModel
 
 PRETEND_N_OF_GPUS = 16

tests/models/test_restore.py

@@ -8,7 +8,7 @@
 import tests.base.utils as tutils
 from pytorch_lightning import Trainer
 from pytorch_lightning.callbacks import ModelCheckpoint
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 from tests.base import (
     LightningTestModel,
     LightningTestModelWithoutHyperparametersArg,

tests/trainer/test_dataloaders.py

@@ -2,7 +2,7 @@
 
 import tests.base.utils as tutils
 from pytorch_lightning import Trainer
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 from tests.base import (
     TestModelBase,
     LightningTestModel,

tests/trainer/test_trainer.py

@@ -14,7 +14,7 @@
 )
 from pytorch_lightning.core.lightning import load_hparams_from_tags_csv
 from pytorch_lightning.trainer.logging import TrainerLoggingMixin
-from pytorch_lightning.utilities.debugging import MisconfigurationException
+from pytorch_lightning.debugging import MisconfigurationException
 from tests.base import (
     TestModelBase,
     DictHparamsModel,