Skip to content

Commit

Permalink
update failing evaluation test to use truncnorm
Browse files Browse the repository at this point in the history
  • Loading branch information
rwedge committed Oct 6, 2023
1 parent 512208d commit b95609b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/integration/evaluation/test_single_table.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,13 +12,13 @@ def test_evaluation():
data = pd.DataFrame({'col': [1, 2, 3]})
metadata = SingleTableMetadata()
metadata.add_column('col', sdtype='numerical')
synthesizer = GaussianCopulaSynthesizer(metadata)
synthesizer = GaussianCopulaSynthesizer(metadata, default_distribution='truncnorm')

# Run and Assert
synthesizer.fit(data)
samples = synthesizer.sample(10)
score = evaluate_quality(data, samples, metadata).get_score()
assert score == 0.8333333333333334
assert score == 0.8666666666666667

diagnostic = run_diagnostic(data, samples, metadata).get_results()
assert diagnostic == {
Expand Down

0 comments on commit b95609b

Please sign in to comment.