diff --git a/src/lightning/app/cli/connect/data.py b/src/lightning/app/cli/connect/data.py index 1c9efcd1f5102..d4d4af2db73f1 100644 --- a/src/lightning/app/cli/connect/data.py +++ b/src/lightning/app/cli/connect/data.py @@ -16,7 +16,6 @@ import click import rich -from lightning_cloud.openapi import ProjectIdDataConnectionsBody from rich.live import Live from rich.spinner import Spinner from rich.text import Text @@ -43,6 +42,8 @@ def connect_data( ) -> None: """Create a new data connection.""" + from lightning_cloud.openapi import ProjectIdDataConnectionsBody + if sys.platform == "win32": _error_and_exit("Data connection isn't supported on windows. Open an issue on Github.")