.

issue.py

@@ -1,3 +1,4 @@
+import cloudpickle
 import pandas as pd
 from sdgym.benchmark import benchmark_single_table
 from sdgym.synthesizers.generate import create_single_table_synthesizer
@@ -19,16 +20,16 @@ def get_trained_synth_v2(data, metadata):
 def sample_synth(trained_synthesizer, num_samples):
     return trained_synthesizer.sample(num_samples)
 
+custom_synthesizer = create_single_table_synthesizer('SimpleGaussianCopula', get_trained_synth, sample_synth)
+custom_synthesizer_v2 = create_single_table_synthesizer('SimpleGaussianCopulaV2', get_trained_synth_v2, sample_synth)
+
 if __name__ == '__main__':
-    custom_synthesizer = create_single_table_synthesizer('SimpleGaussianCopula', get_trained_synth, sample_synth)
-    custom_synthesizer_v2 = create_single_table_synthesizer('SimpleGaussianCopulaV2', get_trained_synth_v2, sample_synth)
     output = benchmark_single_table(
         synthesizers=[],
         sdv_datasets=['fake_hotel_guests'],
         timeout=120,
         sdmetrics=[],
         custom_synthesizers=[custom_synthesizer, custom_synthesizer_v2],
     )
-
     with pd.option_context('display.max_columns', None):
         print(output)