Skip to content

Commit 37bec3c

Browse files
authored
Merge pull request #385 from till-m/colorama
Add `colorama` to fix colors on windows
2 parents 0218b31 + 9a6a2d5 commit 37bec3c

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

bayes_opt/util.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import numpy as np
33
from scipy.stats import norm
44
from scipy.optimize import minimize
5+
from colorama import just_fix_windows_console
56

67

78
def acq_max(ac, gp, y_max, bounds, random_state, constraint=None, n_warmup=10000, n_iter=10):
@@ -295,3 +296,6 @@ def underline(cls, s):
295296
def yellow(cls, s):
296297
"""Wrap text in yellow."""
297298
return cls._wrap_colour(s, cls.YELLOW)
299+
300+
301+
just_fix_windows_console()

setup.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
"numpy >= 1.9.0",
1515
"scipy >= 1.0.0",
1616
"scikit-learn >= 0.18.0",
17+
"colorama"
1718
],
1819
classifiers=[
1920
'License :: OSI Approved :: MIT License',

0 commit comments

Comments
 (0)