Skip to content
Merged
Show file tree
Hide file tree
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions .github/workflows/integration.yml
Original file line number Diff line number Diff line change
Expand Up @@ -10,13 +10,13 @@ jobs:
runs-on: ${{ matrix.os }}
strategy:
matrix:
python-version: ['3.8', '3.9', '3.10', '3.11','3.12']
python-version: ['3.8', '3.9', '3.10', '3.11', '3.12', '3.13']
os: [ubuntu-latest, windows-latest]
include:
- os: macos-latest
python-version: '3.8'
- os: macos-latest
python-version: '3.12'
python-version: '3.13'
steps:
- uses: actions/checkout@v4
- name: Set up Python ${{ matrix.python-version }}
Expand All @@ -30,7 +30,7 @@ jobs:
- name: Run integration tests
run: invoke integration

- if: matrix.os == 'ubuntu-latest' && matrix.python-version == 3.12
- if: matrix.os == 'ubuntu-latest' && matrix.python-version == 3.13
name: Upload integration codecov report
uses: codecov/codecov-action@v4
with:
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/minimum.yml
Original file line number Diff line number Diff line change
Expand Up @@ -11,13 +11,13 @@ jobs:
runs-on: ${{ matrix.os }}
strategy:
matrix:
python-version: ['3.8', '3.9', '3.10', '3.11','3.12']
python-version: ['3.8', '3.9', '3.10', '3.11', '3.12', '3.13']
os: [ubuntu-latest, windows-latest]
include:
- os: macos-13
python-version: '3.8'
- os: macos-latest
python-version: '3.12'
python-version: '3.13'
steps:
- uses: actions/checkout@v4
- name: Set up Python ${{ matrix.python-version }}
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/readme.yml
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ jobs:
runs-on: ${{ matrix.os }}
strategy:
matrix:
python-version: ['3.8', '3.9', '3.10', '3.11','3.12']
python-version: ['3.8', '3.9', '3.10', '3.11', '3.12', '3.13']
os: [ubuntu-latest, macos-latest] # skip windows bc rundoc fails
steps:
- uses: actions/checkout@v4
Expand Down
6 changes: 3 additions & 3 deletions .github/workflows/unit.yml
Original file line number Diff line number Diff line change
Expand Up @@ -10,13 +10,13 @@ jobs:
runs-on: ${{ matrix.os }}
strategy:
matrix:
python-version: ['3.8', '3.9', '3.10', '3.11','3.12']
python-version: ['3.8', '3.9', '3.10', '3.11', '3.12', '3.13']
os: [ubuntu-latest, windows-latest]
include:
- os: macos-latest
python-version: '3.8'
- os: macos-latest
python-version: '3.12'
python-version: '3.13'
steps:
- uses: actions/checkout@v4
- name: Set up Python ${{ matrix.python-version }}
Expand All @@ -30,7 +30,7 @@ jobs:
- name: Run unit tests
run: invoke unit

- if: matrix.os == 'ubuntu-latest' && matrix.python-version == 3.12
- if: matrix.os == 'ubuntu-latest' && matrix.python-version == 3.13
name: Upload unit codecov report
uses: codecov/codecov-action@v4
with:
Expand Down
2 changes: 1 addition & 1 deletion ctgan/synthesizers/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,7 @@ def save(self, path):
def load(cls, path):
"""Load the model stored in the passed `path`."""
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
model = torch.load(path)
model = torch.load(path, weights_only=False)
model.set_device(device)
return model

Expand Down
17 changes: 10 additions & 7 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -13,26 +13,29 @@ classifiers = [
'Programming Language :: Python :: 3.10',
'Programming Language :: Python :: 3.11',
'Programming Language :: Python :: 3.12',
'Programming Language :: Python :: 3.13',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
]
keywords = ['ctgan', 'CTGAN']
dynamic = ['version']
license = { text = 'BSL-1.1' }
requires-python = '>=3.8,<3.13'
requires-python = '>=3.8,<3.14'
readme = 'README.md'
dependencies = [
"numpy>=1.21.0;python_version<'3.10'",
"numpy>=1.23.3;python_version>='3.10' and python_version<'3.12'",
"numpy>=1.26.0;python_version>='3.12'",
"numpy>=1.26.0;python_version>='3.12' and python_version<'3.13'",
"numpy>=2.1.0;python_version>='3.13'",
"pandas>=1.4.0;python_version<'3.11'",
"pandas>=1.5.0;python_version>='3.11' and python_version<'3.12'",
"pandas>=2.1.1;python_version>='3.12'",
"torch>=1.9.0;python_version<'3.10'",
"torch>=1.11.0;python_version>='3.10' and python_version<'3.11'",
"pandas>=2.1.1;python_version>='3.12' and python_version<'3.13'",
"pandas>=2.2.3;python_version>='3.13'",
"torch>=1.13.0;python_version<'3.11'",
"torch>=2.0.0;python_version>='3.11' and python_version<'3.12'",
"torch>=2.2.0;python_version>='3.12'",
"torch>=2.2.0;python_version>='3.12' and python_version<'3.13'",
"torch>=2.6.0;python_version>='3.13'",
'tqdm>=4.29,<5',
'rdt>=1.11.0',
'rdt @ git+https://github.com/sdv-dev/RDT@main',
]

[project.urls]
Expand Down
2 changes: 1 addition & 1 deletion tox.ini
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
[tox]
envlist = py39-lint, py3{8,9,10,11,12}-{unit,integration,readme}
envlist = py39-lint, py3{8,9,10,11,12,13}-{unit,integration,readme}

[testenv]
skipsdist = false
Expand Down
Loading