From e465101e92648fdaedbb373ea04a54238fdd5f5a Mon Sep 17 00:00:00 2001 From: ravi-kumar-pilla Date: Thu, 16 May 2024 20:14:43 -0500 Subject: [PATCH] fix lint Signed-off-by: ravi-kumar-pilla --- package/kedro_viz/api/rest/requests.py | 2 +- package/kedro_viz/api/rest/responses.py | 2 +- package/kedro_viz/api/rest/router.py | 2 +- package/kedro_viz/models/flowchart.py | 9 +++++---- 4 files changed, 8 insertions(+), 7 deletions(-) diff --git a/package/kedro_viz/api/rest/requests.py b/package/kedro_viz/api/rest/requests.py index 373d966cc3..a6ce949d3d 100644 --- a/package/kedro_viz/api/rest/requests.py +++ b/package/kedro_viz/api/rest/requests.py @@ -7,6 +7,6 @@ class DeployerConfiguration(BaseModel): """Credentials for Deployers.""" platform: str - is_preview_enabled: bool + is_datasets_previewed: bool endpoint: str bucket_name: str diff --git a/package/kedro_viz/api/rest/responses.py b/package/kedro_viz/api/rest/responses.py index 1d70f99844..46165a79c8 100644 --- a/package/kedro_viz/api/rest/responses.py +++ b/package/kedro_viz/api/rest/responses.py @@ -482,4 +482,4 @@ def save_api_responses_to_fs(path: str, remote_fs: Any, is_datasets_previewed: b logger.exception( "An error occurred while preparing data for saving. Error: %s", str(exc) ) - raise exc \ No newline at end of file + raise exc diff --git a/package/kedro_viz/api/rest/router.py b/package/kedro_viz/api/rest/router.py index 4de98bf67d..c72003eb60 100644 --- a/package/kedro_viz/api/rest/router.py +++ b/package/kedro_viz/api/rest/router.py @@ -63,7 +63,7 @@ async def deploy_kedro_viz(input_values: DeployerConfiguration): deployer = DeployerFactory.create_deployer( input_values.platform, input_values.endpoint, input_values.bucket_name ) - deployer.deploy(input_values.is_preview_enabled) + deployer.deploy(input_values.is_datasets_previewed) response = { "message": "Website deployed on " f"{input_values.platform and input_values.platform.upper()}", diff --git a/package/kedro_viz/models/flowchart.py b/package/kedro_viz/models/flowchart.py index 4c7984d1e2..45fd1dfdad 100644 --- a/package/kedro_viz/models/flowchart.py +++ b/package/kedro_viz/models/flowchart.py @@ -839,10 +839,11 @@ def set_preview(cls, _): @field_validator("preview_type") @classmethod def set_preview_type(cls, _): - if not cls._is_preview_enabled: - return None - - if cls.data_node.is_preview_disabled() or not hasattr(cls.dataset, "preview"): + if ( + cls.data_node.is_preview_disabled() + or not hasattr(cls.dataset, "preview") + or not cls.is_datasets_previewed + ): return None try: