diff --git a/synth-methods/CTGAN/ctgan_main.py b/synth-methods/CTGAN/ctgan_main.py index 1b2c4e21..1775aef4 100644 --- a/synth-methods/CTGAN/ctgan_main.py +++ b/synth-methods/CTGAN/ctgan_main.py @@ -7,7 +7,7 @@ import sys try: - from ctgan.synthesizer import CTGANSynthesizer + from ctgan import CTGANSynthesizer except ImportError as err: sys.exit("[ERROR] CTGAN library needs to be installed.\nError message: %s" % err)