Skip to content

Commit 36e1acb

Browse files
committed
bump ctgan to 0.11.1 + fix test
1 parent 2bd8841 commit 36e1acb

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ dependencies = [
3535
"pandas>=2.2.3;python_version>='3.13'",
3636
'tqdm>=4.29',
3737
'copulas>=0.12.1',
38-
'ctgan @ git+https://github.com/sdv-dev/CTGAN.git@main',
38+
'ctgan>=0.11.1',
3939
'deepecho>=0.7.0',
4040
'rdt>=1.18.2',
4141
'sdmetrics>=0.21.0',

tests/unit/single_table/test_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -386,7 +386,7 @@ def test__resolve_gpu_parameters(self):
386386
result_without_cuda = instance._resolve_gpu_parameters(parameters_without_cuda)
387387

388388
# Assert
389-
assert result_with_cuda == {'cuda': True, 'enable_gpu': True}
389+
assert result_with_cuda == {'enable_gpu': True}
390390
assert result_with_cuda_only == {'cuda': True}
391391
assert result_with_cuda_none == {'enable_gpu': True}
392392
assert result_without_cuda == {'enable_gpu': False}

0 commit comments

Comments
 (0)