Skip to content

Commit

Permalink
try to fix test failures
Browse files Browse the repository at this point in the history
  • Loading branch information
dholzmueller committed Nov 22, 2024
1 parent 4ef74f6 commit db28c14
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 2 deletions.
5 changes: 3 additions & 2 deletions .github/workflows/testing.yml
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,9 @@ jobs:
python-version: ${{ matrix.python-version }}
- name: Install UV
run: curl -LsSf https://github.com/astral-sh/uv/releases/latest/download/uv-installer.sh | sh
- name: Add UV to path
run: source $HOME/.local/bin/env || echo
- name: Add UV to PATH on Windows
if: "startsWith(runner.os, 'windows')"
run: echo "::add-path::/c/Users/runneradmin/.local/bin"
- name: Install hatch
run: uv pip install --system hatch
- name: Install swig
Expand Down
10 changes: 10 additions & 0 deletions pytabkit/models/sklearn/sklearn_interfaces.py
Original file line number Diff line number Diff line change
Expand Up @@ -1121,6 +1121,11 @@ def _supports_single_sample(self) -> bool:
def _supports_multioutput(self) -> bool:
return False

def _more_tags(self):
# set non-deterministic
# since this class can otherwise fail the check_methods_subset_invariance test due to low precision (?)
return utils.join_dicts(super()._more_tags(), dict(non_deterministic=True))


class FTTransformerConstructorMixin:
def __init__(self,
Expand Down Expand Up @@ -1369,6 +1374,11 @@ def _supports_single_sample(self) -> bool:
def _supports_multioutput(self) -> bool:
return False

def _more_tags(self):
# set non-deterministic
# since this class can otherwise fail the check_methods_subset_invariance test due to low precision (?)
return utils.join_dicts(super()._more_tags(), dict(non_deterministic=True))


class TabrConstructorMixin:
def __init__(self,
Expand Down

0 comments on commit db28c14

Please sign in to comment.