bugfix

Author: stefanch

sgdml/__init__.py

@@ -22,7 +22,7 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-__version__ = '0.5.0'
+__version__ = '0.5.1'
 
 MAX_PRINT_WIDTH = 100
 LOG_LEVELNAME_WIDTH = 7  # do not modify

sgdml/cli.py

@@ -79,7 +79,7 @@ def _print_splash(max_memory, max_processes, use_torch):
 
     version_str = __version__
     version_str += (
-        ' ' + ui.yellow_back_str(' Latest: ' + latest_version + ' ')
+        ' ' + ui.color_str(' Latest: ' + latest_version + ' ', fore_color=ui.BLACK, back_color=ui.YELLOW, bold=True)
         if can_update
         else ''
     )
@@ -101,7 +101,7 @@ def _print_splash(max_memory, max_processes, use_torch):
     if can_update:
         print(
             '\n'
-            + ui.yellow_back_str(' UPDATE AVAILABLE ')
+            + ui.color_str(' UPDATE AVAILABLE ', fore_color=ui.BLACK, back_color=ui.YELLOW, bold=True)
             + '\n'
             + '-' * MAX_PRINT_WIDTH
         )