From 21f8e30abd1a4c92d9cf0c0cc1e2eae5cdde7771 Mon Sep 17 00:00:00 2001 From: Rob <62107751+robsdavis@users.noreply.github.com> Date: Tue, 7 Jan 2025 17:03:53 +0000 Subject: [PATCH] add logging on failed tabular goggle import (#316) --- src/synthcity/plugins/generic/plugin_goggle.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/synthcity/plugins/generic/plugin_goggle.py b/src/synthcity/plugins/generic/plugin_goggle.py index 3e982f06..6c5253c2 100644 --- a/src/synthcity/plugins/generic/plugin_goggle.py +++ b/src/synthcity/plugins/generic/plugin_goggle.py @@ -16,6 +16,7 @@ from torch.utils.data import sampler # synthcity absolute +import synthcity.logger as log from synthcity.plugins.core.dataloader import DataLoader from synthcity.plugins.core.distribution import ( CategoricalDistribution, @@ -32,8 +33,9 @@ from synthcity.plugins.core.models.tabular_goggle import TabularGoggle module_disabled = False -except ImportError: +except ImportError as e: module_disabled = True + log.critical(f"Error importing TabularGoggle: {e}") class GOGGLEPlugin(Plugin): @@ -89,7 +91,7 @@ def __init__( workspace: Path = Path("workspace"), compress_dataset: bool = False, dataloader_sampler: Optional[sampler.Sampler] = None, - **kwargs: Any + **kwargs: Any, ) -> None: """ .. inheritance-diagram:: synthcity.plugins.generic.plugin_goggle.GOGGLEPlugin