Skip to content

Commit 6d86692

Browse files
committed
fix run workflow
1 parent 6684192 commit 6d86692

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

sdgym/_run_benchmark/run_benchmark.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,17 @@
11
import os
22

3-
from sdgym._run_benchmark import OUTPUT_DESTINATION_AWS, RESULTS_UPLOADED
4-
from sdgym.benchmark import SDV_SINGLE_TABLE_SYNTHESIZERS, benchmark_single_table_aws
3+
import sdgym._run_benchmark as run_benchmark
4+
from sdgym.benchmark import benchmark_single_table_aws
55

66
aws_key = os.getenv('AWS_ACCESS_KEY_ID')
77
aws_secret = os.getenv('AWS_SECRET_ACCESS_KEY')
8-
synthesizer = SDV_SINGLE_TABLE_SYNTHESIZERS
98
datasets = ['expedia_hotel_logs', 'fake_companies']
109

11-
1210
if __name__ == '__main__':
13-
RESULTS_UPLOADED = False
11+
run_benchmark.RESULTS_UPLOADED = False
1412
for synthesizer in ['GaussianCopulaSynthesizer', 'TVAESynthesizer']:
1513
benchmark_single_table_aws(
16-
output_destination=OUTPUT_DESTINATION_AWS,
14+
output_destination=run_benchmark.OUTPUT_DESTINATION_AWS,
1715
aws_access_key_id=aws_key,
1816
aws_secret_access_key=aws_secret,
1917
synthesizers=[synthesizer],

0 commit comments

Comments
 (0)