From 90ee292214425505ebafa3b02a2742f5bc5c7bb7 Mon Sep 17 00:00:00 2001
From: Tobias Pitters <tobias.pitters@gmail.com>
Date: Tue, 3 Jun 2025 11:15:31 +0200
Subject: [PATCH 1/8] add pyproject.toml instead of requirements and setup
 files

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>
---
 .github/workflows/notebook-tests.yml | 10 +--
 .github/workflows/python-linting.yml |  2 +-
 .github/workflows/python-package.yml | 13 ++--
 .github/workflows/python-publish.yml |  6 +-
 pyproject.toml                       | 94 ++++++++++++++++++++++++++++
 requirements-deeplearning.txt        |  2 -
 requirements-linting.txt             | 10 ---
 requirements-test.txt                |  7 ---
 requirements.txt                     |  8 ---
 setup.py                             | 53 ----------------
 10 files changed, 104 insertions(+), 101 deletions(-)
 create mode 100644 pyproject.toml
 delete mode 100644 requirements-deeplearning.txt
 delete mode 100644 requirements-linting.txt
 delete mode 100644 requirements-test.txt
 delete mode 100644 requirements.txt
 delete mode 100644 setup.py

diff --git a/.github/workflows/notebook-tests.yml b/.github/workflows/notebook-tests.yml
index 921d2197..fb24116c 100644
--- a/.github/workflows/notebook-tests.yml
+++ b/.github/workflows/notebook-tests.yml
@@ -29,15 +29,9 @@ jobs:
     - name: Upgrade pip
       run: |
         python -m pip install --upgrade pip
-    - name: Install core dependencies
+    - name: Install all dependencies
       run: |
-        pip install -r requirements.txt
-    - name: Install deep learning dependencies
-      run: |
-        pip install -r requirements-deeplearning.txt
-    - name: Install test dependencies
-      run: |
-        pip install -r requirements-test.txt
+        python -m pip install -e .[deeplearning,test]
     - name: Test with pytest
       run: |
         # pytest
diff --git a/.github/workflows/python-linting.yml b/.github/workflows/python-linting.yml
index 319c147a..acc0d7af 100644
--- a/.github/workflows/python-linting.yml
+++ b/.github/workflows/python-linting.yml
@@ -24,7 +24,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install -r requirements-linting.txt
+        python -m pip install -e .[linting]
     - name: Check sorted python imports using isort
       run: |
         isort . -c
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 9b6fe54c..22007a46 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -29,15 +29,9 @@ jobs:
     - name: Upgrade pip
       run: |
         python -m pip install --upgrade pip
-    - name: Install core dependencies
+    - name: Install all dependencies
       run: |
-        pip install -r requirements.txt
-    - name: Install deep learning dependencies
-      run: |
-        pip install -r requirements-deeplearning.txt
-    - name: Install test dependencies
-      run: |
-        pip install -r requirements-test.txt
+        python -m pip install -e .[deeplearning,test]
     - name: Test with pytest
       run: |
         # pytest
@@ -62,5 +56,6 @@ jobs:
     #     verbose: true
     - name: Check package consistency with twine
       run: |
-        python setup.py check sdist bdist_wheel
+        pip install build twine
+        python -m build
         twine check dist/* 
diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index 15f8f65c..e0627737 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -21,9 +21,9 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install setuptools wheel twine
-    - name: Build and publish
+        pip install build twine
+    - name: Build
       run: |
-        python setup.py sdist bdist_wheel
+        python -m build
     - name: Publish package distributions to PyPI
       uses: pypa/gh-action-pypi-publish@release/v1
diff --git a/pyproject.toml b/pyproject.toml
new file mode 100644
index 00000000..94708ce5
--- /dev/null
+++ b/pyproject.toml
@@ -0,0 +1,94 @@
+[build-system]
+requires = ["setuptools>=42", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "dice_ml"
+version = "0.11"
+authors = [
+    {name = "Ramaravind Mothilal", email = "raam.arvind93@gmail.com"},
+    {name = "Amit Sharma"},
+    {name = "Chenhao Tan"}
+]
+description = "Generate Diverse Counterfactual Explanations for any machine learning model."
+readme = "README.rst"
+requires-python = ">=3.9"
+license = {text = "MIT"}
+classifiers = [
+    "Intended Audience :: Developers",
+    "Intended Audience :: Science/Research",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+keywords = ["machine-learning", "explanation", "interpretability", "counterfactual"]
+dependencies = [
+    "jsonschema",
+    "numpy",
+    "pandas>=2.0.0",
+    "scikit-learn",
+    "tqdm",
+    "raiutils>=0.4.0",
+    "xgboost",
+    "lightgbm",
+]
+
+[project.urls]
+"Homepage" = "https://github.com/interpretml/DiCE"
+"Download" = "https://github.com/interpretml/DiCE/archive/v0.11.tar.gz"
+
+[project.optional-dependencies]
+deeplearning = [
+    "tensorflow>=1.13.1",
+    "torch",
+]
+test = [
+    "ipython",
+    "jupyter",
+    "pytest",
+    "pytest-cov",
+    "twine",
+    "pytest-mock",
+    "rai_test_utils",
+]
+linting = [
+    "flake8",
+    "flake8-bugbear",
+    "flake8-blind-except",
+    "flake8-breakpoint",
+    "flake8-builtins",
+    "flake8-logging-format",
+    "flake8-pytest-style",
+    "flake8-all-not-strings",
+    "isort",
+    "packaging",
+]
+dev = [
+    "ipython",
+    "jupyter",
+    "pytest",
+    "pytest-cov",
+    "twine",
+    "pytest-mock",
+    "rai_test_utils",
+    "flake8",
+    "flake8-bugbear",
+    "flake8-blind-except",
+    "flake8-breakpoint",
+    "flake8-builtins",
+    "flake8-logging-format",
+    "flake8-pytest-style",
+    "flake8-all-not-strings",
+    "isort",
+    "packaging",
+]
+
+[tool.setuptools]
+packages = {find = {exclude = ["tests*"]}}
+include-package-data = true
+
+[tool.setuptools.package-data]
+"*" = ["*.h5", "counterfactual_explanations_v1.0.json", "counterfactual_explanations_v2.0.json"]
\ No newline at end of file
diff --git a/requirements-deeplearning.txt b/requirements-deeplearning.txt
deleted file mode 100644
index 3c0e3c09..00000000
--- a/requirements-deeplearning.txt
+++ /dev/null
@@ -1,2 +0,0 @@
-tensorflow>=1.13.1
-torch
diff --git a/requirements-linting.txt b/requirements-linting.txt
deleted file mode 100644
index d7320de7..00000000
--- a/requirements-linting.txt
+++ /dev/null
@@ -1,10 +0,0 @@
-flake8
-flake8-bugbear
-flake8-blind-except
-flake8-breakpoint
-flake8-builtins
-flake8-logging-format
-flake8-pytest-style
-flake8-all-not-strings
-isort
-packaging
diff --git a/requirements-test.txt b/requirements-test.txt
deleted file mode 100644
index 5a7b5bf3..00000000
--- a/requirements-test.txt
+++ /dev/null
@@ -1,7 +0,0 @@
-ipython
-jupyter
-pytest
-pytest-cov
-twine
-pytest-mock
-rai_test_utils
diff --git a/requirements.txt b/requirements.txt
deleted file mode 100644
index 272e006a..00000000
--- a/requirements.txt
+++ /dev/null
@@ -1,8 +0,0 @@
-jsonschema
-numpy # if you are using tensorflow 1.x, it requires numpy<=1.16 
-pandas>=2.0.0
-scikit-learn
-tqdm
-raiutils>=0.4.0
-xgboost # if you are using xgboost
-lightgbm # if you are using lightgbm
\ No newline at end of file
diff --git a/setup.py b/setup.py
deleted file mode 100644
index edaba5f7..00000000
--- a/setup.py
+++ /dev/null
@@ -1,53 +0,0 @@
-import setuptools
-
-VERSION_STR = "0.11"
-
-with open("README.rst", "r") as fh:
-    long_description = fh.read()
-
-# Get the required packages
-with open('requirements.txt', encoding='utf-8') as f:
-    install_requires = f.read().splitlines()
-
-# Deep learning packages are optional to install
-extras = ["deeplearning"]
-extras_require = dict()
-for e in extras:
-    req_file = "requirements-{0}.txt".format(e)
-    with open(req_file) as f:
-        extras_require[e] = [line.strip() for line in f]
-
-setuptools.setup(
-    name="dice_ml",
-    version=VERSION_STR,
-    license="MIT",
-    author="Ramaravind Mothilal, Amit Sharma, Chenhao Tan",
-    author_email="raam.arvind93@gmail.com",
-    description="Generate Diverse Counterfactual Explanations for any machine learning model.",
-    long_description=long_description,
-    long_description_content_type="text/x-rst",
-    url="https://github.com/interpretml/DiCE",
-    download_url="https://github.com/interpretml/DiCE/archive/v"+VERSION_STR+".tar.gz",
-    python_requires='>=3.9',
-    packages=setuptools.find_packages(exclude=['tests*']),
-    classifiers=[
-        "Intended Audience :: Developers",
-        "Intended Audience :: Science/Research",
-        "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.9",
-        "Programming Language :: Python :: 3.10",
-        "Programming Language :: Python :: 3.11",
-        "License :: OSI Approved :: MIT License",
-        "Operating System :: OS Independent",
-    ],
-    keywords='machine-learning explanation interpretability counterfactual',
-    install_requires=install_requires,
-    extras_require=extras_require,
-    include_package_data=True,
-    package_data={
-        # If any package contains *.h5 files, include them:
-        '': ['*.h5',
-             'counterfactual_explanations_v1.0.json',
-             'counterfactual_explanations_v2.0.json']
-    }
-)

From e2e8dfced7af23e49df668cd143e49371befc12f Mon Sep 17 00:00:00 2001
From: Tobias Pitters <tobias.pitters@gmail.com>
Date: Tue, 3 Jun 2025 11:53:34 +0200
Subject: [PATCH 2/8] update workflows

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>
---
 .github/workflows/notebook-tests.yml | 7 ++++---
 .github/workflows/python-linting.yml | 7 +++++--
 .github/workflows/python-package.yml | 7 ++++---
 .github/workflows/python-publish.yml | 7 +++++--
 pyproject.toml                       | 4 ++--
 5 files changed, 20 insertions(+), 12 deletions(-)

diff --git a/.github/workflows/notebook-tests.yml b/.github/workflows/notebook-tests.yml
index fb24116c..77767d05 100644
--- a/.github/workflows/notebook-tests.yml
+++ b/.github/workflows/notebook-tests.yml
@@ -26,12 +26,13 @@ jobs:
       uses: actions/setup-python@v5
       with:
         python-version: ${{ matrix.python-version }}
-    - name: Upgrade pip
+    - name: Install uv
       run: |
-        python -m pip install --upgrade pip
+        curl -LsSf https://astral.sh/uv/install.sh | sh
+        echo "$HOME/.cargo/bin" >> $GITHUB_PATH
     - name: Install all dependencies
       run: |
-        python -m pip install -e .[deeplearning,test]
+        uv pip install -e .[deeplearning,test]
     - name: Test with pytest
       run: |
         # pytest
diff --git a/.github/workflows/python-linting.yml b/.github/workflows/python-linting.yml
index acc0d7af..738fe49d 100644
--- a/.github/workflows/python-linting.yml
+++ b/.github/workflows/python-linting.yml
@@ -21,10 +21,13 @@ jobs:
       uses: actions/setup-python@v5
       with:
         python-version: '3.11'
+    - name: Install uv
+      run: |
+        curl -LsSf https://astral.sh/uv/install.sh | sh
+        echo "$HOME/.cargo/bin" >> $GITHUB_PATH
     - name: Install dependencies
       run: |
-        python -m pip install --upgrade pip
-        python -m pip install -e .[linting]
+        uv pip install -e .[linting]
     - name: Check sorted python imports using isort
       run: |
         isort . -c
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 22007a46..94cfbb88 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -26,12 +26,13 @@ jobs:
       uses: actions/setup-python@v5
       with:
         python-version: ${{ matrix.python-version }}
-    - name: Upgrade pip
+    - name: Install uv
       run: |
-        python -m pip install --upgrade pip
+        curl -LsSf https://astral.sh/uv/install.sh | sh
+        echo "$HOME/.cargo/bin" >> $GITHUB_PATH
     - name: Install all dependencies
       run: |
-        python -m pip install -e .[deeplearning,test]
+        uv pip install -e .[deeplearning,test]
     - name: Test with pytest
       run: |
         # pytest
diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index e0627737..9f6ada2e 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -18,10 +18,13 @@ jobs:
       uses: actions/setup-python@v5
       with:
         python-version: '3.x'
+    - name: Install uv
+      run: |
+        curl -LsSf https://astral.sh/uv/install.sh | sh
+        echo "$HOME/.cargo/bin" >> $GITHUB_PATH
     - name: Install dependencies
       run: |
-        python -m pip install --upgrade pip
-        pip install build twine
+        uv pip install build twine
     - name: Build
       run: |
         python -m build
diff --git a/pyproject.toml b/pyproject.toml
index 94708ce5..943bc02a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dice_ml"
-version = "0.11"
+dynamic = ["version"]
 authors = [
     {name = "Ramaravind Mothilal", email = "raam.arvind93@gmail.com"},
     {name = "Amit Sharma"},
@@ -91,4 +91,4 @@ packages = {find = {exclude = ["tests*"]}}
 include-package-data = true
 
 [tool.setuptools.package-data]
-"*" = ["*.h5", "counterfactual_explanations_v1.0.json", "counterfactual_explanations_v2.0.json"]
\ No newline at end of file
+"*" = ["*.h5", "counterfactual_explanations_v1.0.json", "counterfactual_explanations_v2.0.json"]

From f5608b198b7ac09da2c016b316930776c02b89a7 Mon Sep 17 00:00:00 2001
From: Tobias Pitters <tobias.pitters@gmail.com>
Date: Tue, 3 Jun 2025 11:55:07 +0200
Subject: [PATCH 3/8] update workflows

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>
---
 .github/workflows/notebook-tests.yml | 2 +-
 .github/workflows/python-linting.yml | 2 +-
 .github/workflows/python-package.yml | 2 +-
 .github/workflows/python-publish.yml | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/notebook-tests.yml b/.github/workflows/notebook-tests.yml
index 77767d05..b6d15949 100644
--- a/.github/workflows/notebook-tests.yml
+++ b/.github/workflows/notebook-tests.yml
@@ -32,7 +32,7 @@ jobs:
         echo "$HOME/.cargo/bin" >> $GITHUB_PATH
     - name: Install all dependencies
       run: |
-        uv pip install -e .[deeplearning,test]
+        uv pip install --system -e .[deeplearning,test]
     - name: Test with pytest
       run: |
         # pytest
diff --git a/.github/workflows/python-linting.yml b/.github/workflows/python-linting.yml
index 738fe49d..c5bb24e4 100644
--- a/.github/workflows/python-linting.yml
+++ b/.github/workflows/python-linting.yml
@@ -27,7 +27,7 @@ jobs:
         echo "$HOME/.cargo/bin" >> $GITHUB_PATH
     - name: Install dependencies
       run: |
-        uv pip install -e .[linting]
+        uv pip install --system -e .[linting]
     - name: Check sorted python imports using isort
       run: |
         isort . -c
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 94cfbb88..6e020606 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -32,7 +32,7 @@ jobs:
         echo "$HOME/.cargo/bin" >> $GITHUB_PATH
     - name: Install all dependencies
       run: |
-        uv pip install -e .[deeplearning,test]
+        uv pip install --system -e .[deeplearning,test]
     - name: Test with pytest
       run: |
         # pytest
diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index 9f6ada2e..d373cd7e 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -24,7 +24,7 @@ jobs:
         echo "$HOME/.cargo/bin" >> $GITHUB_PATH
     - name: Install dependencies
       run: |
-        uv pip install build twine
+        uv pip install --system build twine
     - name: Build
       run: |
         python -m build

From 97dfa940c9370182ee3d9421ad54c432e6a4aa17 Mon Sep 17 00:00:00 2001
From: Tobias Pitters <tobias.pitters@gmail.com>
Date: Tue, 3 Jun 2025 11:59:29 +0200
Subject: [PATCH 4/8] update things

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>
---
 .github/workflows/notebook-tests.yml | 3 +--
 .github/workflows/python-linting.yml | 3 +--
 .github/workflows/python-package.yml | 3 +--
 .github/workflows/python-publish.yml | 3 +--
 4 files changed, 4 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/notebook-tests.yml b/.github/workflows/notebook-tests.yml
index b6d15949..c34df20c 100644
--- a/.github/workflows/notebook-tests.yml
+++ b/.github/workflows/notebook-tests.yml
@@ -28,8 +28,7 @@ jobs:
         python-version: ${{ matrix.python-version }}
     - name: Install uv
       run: |
-        curl -LsSf https://astral.sh/uv/install.sh | sh
-        echo "$HOME/.cargo/bin" >> $GITHUB_PATH
+        pip install uv
     - name: Install all dependencies
       run: |
         uv pip install --system -e .[deeplearning,test]
diff --git a/.github/workflows/python-linting.yml b/.github/workflows/python-linting.yml
index c5bb24e4..e0de417f 100644
--- a/.github/workflows/python-linting.yml
+++ b/.github/workflows/python-linting.yml
@@ -23,8 +23,7 @@ jobs:
         python-version: '3.11'
     - name: Install uv
       run: |
-        curl -LsSf https://astral.sh/uv/install.sh | sh
-        echo "$HOME/.cargo/bin" >> $GITHUB_PATH
+        pip install uv
     - name: Install dependencies
       run: |
         uv pip install --system -e .[linting]
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 6e020606..465f5b71 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -28,8 +28,7 @@ jobs:
         python-version: ${{ matrix.python-version }}
     - name: Install uv
       run: |
-        curl -LsSf https://astral.sh/uv/install.sh | sh
-        echo "$HOME/.cargo/bin" >> $GITHUB_PATH
+        pip install uv
     - name: Install all dependencies
       run: |
         uv pip install --system -e .[deeplearning,test]
diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index d373cd7e..7c86f5ea 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -20,8 +20,7 @@ jobs:
         python-version: '3.x'
     - name: Install uv
       run: |
-        curl -LsSf https://astral.sh/uv/install.sh | sh
-        echo "$HOME/.cargo/bin" >> $GITHUB_PATH
+        pip install uv
     - name: Install dependencies
       run: |
         uv pip install --system build twine

From 93ee3fb2f61fab6227580226414ebcee9da84a4c Mon Sep 17 00:00:00 2001
From: Tobias Pitters <tobias.pitters@gmail.com>
Date: Tue, 3 Jun 2025 15:00:17 +0200
Subject: [PATCH 5/8] update python package flow

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>
---
 .github/workflows/python-package.yml | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 465f5b71..179cf276 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -17,8 +17,10 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        python-version: ["3.9", "3.10", "3.11", "3.12"]
-        os: [ubuntu-latest, macos-latest, windows-latest]
+        # python-version: ["3.9", "3.10", "3.11", "3.12"]
+        python-version: ["3.10"]
+        # os: [ubuntu-latest, macos-latest, windows-latest]
+        os: [ubuntu-latest]
 
     steps:
     - uses: actions/checkout@v4
@@ -32,6 +34,8 @@ jobs:
     - name: Install all dependencies
       run: |
         uv pip install --system -e .[deeplearning,test]
+    - name: Setup tmate session
+      uses: mxschmitt/action-tmate@v3
     - name: Test with pytest
       run: |
         # pytest

From 250de8b03461e7e019676fe65b750722060775d0 Mon Sep 17 00:00:00 2001
From: Tobias Pitters <tobias.pitters@gmail.com>
Date: Tue, 3 Jun 2025 15:17:00 +0200
Subject: [PATCH 6/8] WIP: example test run

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>
---
 pyproject.toml     | 5 +++--
 tests/test_data.py | 1 +
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 943bc02a..50746238 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dice_ml"
-dynamic = ["version"]
+version = "0.11"
 authors = [
     {name = "Ramaravind Mothilal", email = "raam.arvind93@gmail.com"},
     {name = "Amit Sharma"},
@@ -27,7 +27,7 @@ classifiers = [
 keywords = ["machine-learning", "explanation", "interpretability", "counterfactual"]
 dependencies = [
     "jsonschema",
-    "numpy",
+    "numpy>=1.20.0",
     "pandas>=2.0.0",
     "scikit-learn",
     "tqdm",
@@ -42,6 +42,7 @@ dependencies = [
 
 [project.optional-dependencies]
 deeplearning = [
+    "numpy<2.0.0",
     "tensorflow>=1.13.1",
     "torch",
 ]
diff --git a/tests/test_data.py b/tests/test_data.py
index d562a5aa..746a9010 100644
--- a/tests/test_data.py
+++ b/tests/test_data.py
@@ -71,6 +71,7 @@ def test_ohe_min_max_transformed_query_instance(self, sample_adultincome_query):
         output_query = [0.068, 0.449, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0,
                         0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0, 1.0, 0.0]
         d = self.d[0]
+        import ipdb; ipdb.set_trace(context=20)
         prepared_query = d.get_ohe_min_max_normalized_data(query_instance=sample_adultincome_query).iloc[0].tolist()
         assert output_query == pytest.approx(prepared_query, abs=1e-3)
 

From 769b459f18471e1f2fb8faa83c9144b088f00d07 Mon Sep 17 00:00:00 2001
From: Tobias Pitters <tobias.pitters@gmail.com>
Date: Tue, 3 Jun 2025 15:34:03 +0200
Subject: [PATCH 7/8] fix tests

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>
---
 .github/workflows/python-package.yml | 6 ++----
 pyproject.toml                       | 4 ++--
 tests/test_data.py                   | 1 -
 3 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 179cf276..7269fdc6 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -17,10 +17,8 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        # python-version: ["3.9", "3.10", "3.11", "3.12"]
-        python-version: ["3.10"]
-        # os: [ubuntu-latest, macos-latest, windows-latest]
-        os: [ubuntu-latest]
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
+        os: [ubuntu-latest, macos-latest, windows-latest]
 
     steps:
     - uses: actions/checkout@v4
diff --git a/pyproject.toml b/pyproject.toml
index 50746238..62263d83 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -49,7 +49,7 @@ deeplearning = [
 test = [
     "ipython",
     "jupyter",
-    "pytest",
+    "pytest==8.3.5",
     "pytest-cov",
     "twine",
     "pytest-mock",
@@ -70,7 +70,7 @@ linting = [
 dev = [
     "ipython",
     "jupyter",
-    "pytest",
+    "pytest==8.3.5",
     "pytest-cov",
     "twine",
     "pytest-mock",
diff --git a/tests/test_data.py b/tests/test_data.py
index 746a9010..d562a5aa 100644
--- a/tests/test_data.py
+++ b/tests/test_data.py
@@ -71,7 +71,6 @@ def test_ohe_min_max_transformed_query_instance(self, sample_adultincome_query):
         output_query = [0.068, 0.449, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0,
                         0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0, 1.0, 0.0]
         d = self.d[0]
-        import ipdb; ipdb.set_trace(context=20)
         prepared_query = d.get_ohe_min_max_normalized_data(query_instance=sample_adultincome_query).iloc[0].tolist()
         assert output_query == pytest.approx(prepared_query, abs=1e-3)
 

From bbde3f5f230bfea1e84bb17b542120245bb2b5e4 Mon Sep 17 00:00:00 2001
From: Tobias Pitters <tobias.pitters@gmail.com>
Date: Tue, 3 Jun 2025 15:48:45 +0200
Subject: [PATCH 8/8] remove debugging from workflow

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>
---
 .github/workflows/python-package.yml | 2 --
 1 file changed, 2 deletions(-)

diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 7269fdc6..465f5b71 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -32,8 +32,6 @@ jobs:
     - name: Install all dependencies
       run: |
         uv pip install --system -e .[deeplearning,test]
-    - name: Setup tmate session
-      uses: mxschmitt/action-tmate@v3
     - name: Test with pytest
       run: |
         # pytest