diff --git a/pytorch_lightning/plugins/__init__.py b/pytorch_lightning/plugins/__init__.py index e69de29bb2d1d..e023060d5b16a 100644 --- a/pytorch_lightning/plugins/__init__.py +++ b/pytorch_lightning/plugins/__init__.py @@ -0,0 +1,3 @@ +from pytorch_lightning.plugins.base_plugin import Plugin +from pytorch_lightning.plugins.precision import * +from pytorch_lightning.plugins.training_type import * diff --git a/pytorch_lightning/plugins/precision/__init__.py b/pytorch_lightning/plugins/precision/__init__.py index e69de29bb2d1d..e023060d5b16a 100644 --- a/pytorch_lightning/plugins/precision/__init__.py +++ b/pytorch_lightning/plugins/precision/__init__.py @@ -0,0 +1,3 @@ +from pytorch_lightning.plugins.base_plugin import Plugin +from pytorch_lightning.plugins.precision import * +from pytorch_lightning.plugins.training_type import * diff --git a/pytorch_lightning/plugins/training_type/__init__.py b/pytorch_lightning/plugins/training_type/__init__.py index e69de29bb2d1d..329f6347b17c3 100644 --- a/pytorch_lightning/plugins/training_type/__init__.py +++ b/pytorch_lightning/plugins/training_type/__init__.py @@ -0,0 +1 @@ +from pytorch_lightning.plugins.training_type.training_type_plugin import TrainingTypePlugin