diff --git a/setup.py b/setup.py index 519f495c3b041..7d4084960d450 100755 --- a/setup.py +++ b/setup.py @@ -55,7 +55,7 @@ _PACKAGE_NAME = os.environ.get("PACKAGE_NAME", "") _PACKAGE_MAPPING = {"pytorch": "pytorch_lightning", "app": "lightning_app"} -_REAL_PKG_NAME = _PACKAGE_MAPPING.get(_PACKAGE_NAME, "") +_REAL_PKG_NAME = _PACKAGE_MAPPING.get(_PACKAGE_NAME, _PACKAGE_NAME) # https://packaging.python.org/guides/single-sourcing-package-version/ # http://blog.ionelmc.ro/2014/05/25/python-packaging/ _PATH_ROOT = os.path.dirname(__file__)