.

Lightning-AI · Aug 15, 2022 · 4d1454d · 4d1454d
1 parent e526466
commit 4d1454d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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__)