Merge pull request #385 from till-m/colorama

Add `colorama` to fix colors on windows
bayesian-optimization · Nov 29, 2022 · 37bec3c · 37bec3c
2 parents 0218b31 + 9a6a2d5
commit 37bec3c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/bayes_opt/util.py b/bayes_opt/util.py
@@ -2,6 +2,7 @@
 import numpy as np
 from scipy.stats import norm
 from scipy.optimize import minimize
+from colorama import just_fix_windows_console
 
 
 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):
     def yellow(cls, s):
         """Wrap text in yellow."""
         return cls._wrap_colour(s, cls.YELLOW)
+
+
+just_fix_windows_console()
diff --git a/setup.py b/setup.py
@@ -14,6 +14,7 @@
         "numpy >= 1.9.0",
         "scipy >= 1.0.0",
         "scikit-learn >= 0.18.0",
+        "colorama"
     ],
     classifiers=[
         'License :: OSI Approved :: MIT License',