add logging on failed tabular goggle import (#316)

vanderschaarlab · Jan 7, 2025 · 21f8e30 · 21f8e30
1 parent d6da33a
commit 21f8e30
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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