File tree Expand file tree Collapse file tree 1 file changed +4
-6
lines changed
Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Original file line number Diff line number Diff line change 11import 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
66aws_key = os .getenv ('AWS_ACCESS_KEY_ID' )
77aws_secret = os .getenv ('AWS_SECRET_ACCESS_KEY' )
8- synthesizer = SDV_SINGLE_TABLE_SYNTHESIZERS
98datasets = ['expedia_hotel_logs' , 'fake_companies' ]
109
11-
1210if __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 ],
You can’t perform that action at this time.
0 commit comments