Skip to content

Commit ce27ba2

Browse files
authored
Quantization type hint bug fix (#350)
* Fix inconsistent type hints for Float hyperparameter quantization parameter Fixes #349 * ignore virtual environments
1 parent ed74a51 commit ce27ba2

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

.gitignore

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,5 +70,8 @@ prof/
7070
# VSCode
7171
.vscode
7272

73+
# Virtual environments
74+
.venv
75+
7376
# Running pre-commit seems to generate these
7477
.mypy_cache

ConfigSpace/api/types/float.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def Float(
1818
*,
1919
distribution: Uniform | None = ...,
2020
default: float | None = ...,
21-
q: int | None = ...,
21+
q: int | float | None = ...,
2222
log: bool = ...,
2323
meta: dict | None = ...,
2424
) -> UniformFloatHyperparameter:
@@ -33,7 +33,7 @@ def Float(
3333
*,
3434
distribution: Normal,
3535
default: float | None = ...,
36-
q: int | None = ...,
36+
q: int | float | None = ...,
3737
log: bool = ...,
3838
meta: dict | None = ...,
3939
) -> NormalFloatHyperparameter:
@@ -113,7 +113,7 @@ def Float(
113113
default : float | None = None
114114
The default value to give to the hyperparameter.
115115
116-
q : float | None = None
116+
q : float | int | None = None
117117
The quantization factor, must evenly divide the boundaries.
118118
119119
Note

0 commit comments

Comments
 (0)