update

Author: R-Palazzo

sdgym/benchmark.py

@@ -242,7 +242,7 @@ def _setup_output_destination(
             The s3 client that can be used to read / write to s3. Defaults to ``None``.
     """
     if s3_client:
-        return _setup_output_destination_aws(output_destination, synthesizers, datasets, s3_client)
+        return _setup_output_destination_aws(output_destination, synthesizers, datasets, modality, s3_client)
 
     if output_destination is None:
         return {}
@@ -399,7 +399,7 @@ def _synthesize(
     synthesizer_obj = get_synthesizer(data, metadata)
     synthesizer_size = len(cloudpickle.dumps(synthesizer_obj)) / N_BYTES_IN_MB
     train_now = get_utc_now()
-    synthetic_data = sample_from_synthesizer(synthesizer_obj, n_samples=num_samples)
+    synthetic_data = sample_from_synthesizer(synthesizer_obj, num_samples)
     sample_now = get_utc_now()
 
     peak_memory = tracemalloc.get_traced_memory()[1] / N_BYTES_IN_MB