Unpin Ray nightly in CI (#2614) (#2646)

.github/workflows/pytest.yml

@@ -114,10 +114,10 @@ jobs:
           if [ "$MARKERS" == "distributed" ]; then
             if [ "$RAY_VERSION" == "nightly" ]; then
               # NOTE: hardcoded for python 3.9 on Linux
-              # pip install https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-3.0.0.dev0-cp39-cp39-manylinux2014_x86_64.whl
+              pip install https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-3.0.0.dev0-cp39-cp39-manylinux2014_x86_64.whl
 
               # NOTE: Pinned Ray nightly version to September 20, 2022 to get tests to pass
-              pip install https://s3-us-west-2.amazonaws.com/ray-wheels/master/fa182d3c9e478ef4c169ccf7459764768996110f/ray-3.0.0.dev0-cp39-cp39-manylinux2014_x86_64.whl
+              # pip install https://s3-us-west-2.amazonaws.com/ray-wheels/master/fa182d3c9e478ef4c169ccf7459764768996110f/ray-3.0.0.dev0-cp39-cp39-manylinux2014_x86_64.whl
             else
               pip install ray==$RAY_VERSION
             fi

ludwig/__init__.py

@@ -18,3 +18,6 @@
 from ludwig.globals import LUDWIG_VERSION as __version__  # noqa
 
 logging.basicConfig(level=logging.INFO, stream=sys.stdout, format="%(message)s")
+
+# Disable annoying message about NUMEXPR_MAX_THREADS
+logging.getLogger("numexpr").setLevel(logging.WARNING)

ludwig/utils/visualization_utils.py

@@ -15,7 +15,6 @@
 # ==============================================================================
 import copy
 import logging
-import sys
 from collections import Counter
 from sys import platform
 
@@ -32,7 +31,10 @@
     import matplotlib as mpl
 
     if platform == "darwin":  # OS X
-        mpl.use("TkAgg")
+        try:
+            mpl.use("TkAgg")
+        except ModuleNotFoundError:
+            logging.warning("Unable to set TkAgg backend for matplotlib. Your Python may not be configured for Tk")
     import matplotlib.patches as patches
     import matplotlib.path as path
     import matplotlib.patheffects as PathEffects
@@ -41,13 +43,12 @@
     from matplotlib import ticker
     from matplotlib.lines import Line2D
     from mpl_toolkits.mplot3d import Axes3D
-except ImportError:
-    logger.error(
-        " matplotlib or seaborn are not installed. "
+except ImportError as e:
+    raise RuntimeError(
+        "matplotlib or seaborn are not installed. "
         "In order to install all visualization dependencies run "
         "pip install ludwig[viz]"
-    )
-    sys.exit(-1)
+    ) from e
 
 INT_QUANTILES = 10
 FLOAT_QUANTILES = 10