address comments

Signed-off-by: Weichen Xu <weichen.xu@databricks.com>

mlflow/models/utils.py

@@ -60,9 +60,13 @@ def _is_scalar(x):
             return np.isscalar(x) or x is None
 
         def _is_tensor(x):
-            return isinstance(x, np.ndarray) or (
-                isinstance(x, dict) and all([isinstance(ary, np.ndarray) for ary in x.values()])
-            ) or isinstance(x, (csr_matrix, csc_matrix))
+            return (
+                isinstance(x, np.ndarray)
+                or (
+                    isinstance(x, dict) and all([isinstance(ary, np.ndarray) for ary in x.values()])
+                )
+                or isinstance(x, (csr_matrix, csc_matrix))
+            )
 
         def _handle_tensor_input(input_tensor: Union[np.ndarray, dict, csr_matrix, csc_matrix]):
             if isinstance(input_tensor, dict):

mlflow/types/utils.py

@@ -31,6 +31,7 @@ def _get_tensor_shape(data: np.ndarray, variable_dimension: Optional[int] = 0) -
     :return: tuple : Shape of the inputted data (including a variable dimension)
     """
     from scipy.sparse import csr_matrix, csc_matrix
+
     if not isinstance(data, (np.ndarray, csr_matrix, csc_matrix)):
         raise TypeError("Expected numpy.ndarray or csc/csr matrix, got '{}'.".format(type(data)))
     variable_input_data_shape = data.shape
@@ -101,6 +102,7 @@ def _infer_schema(data: Any) -> Schema:
     :return: Schema
     """
     from scipy.sparse import csr_matrix, csc_matrix
+
     if isinstance(data, dict):
         res = []
         for name in data.keys():

mlflow/utils/autologging_utils/__init__.py

@@ -174,8 +174,6 @@ def resolve_input_example_and_signature(
             model_signature = infer_model_signature(input_example)
         except Exception as e:
             model_signature_user_msg = "Failed to infer model signature: " + str(e)
-            import traceback
-            traceback.print_exc()
 
     if log_input_example and input_example_user_msg is not None:
         logger.warning(input_example_user_msg)

setup.py

@@ -71,6 +71,7 @@ def package_files(directory):
     "Flask",
     "gunicorn; platform_system != 'Windows'",
     "numpy",
+    "scipy",
     "pandas",
     "prometheus-flask-exporter",
     "querystring_parser",