Merge branch 'zwei' into fix-serde-compatability

laurenyu · web-flow · commit 34db6068c5bd · 2020-07-28T17:57:03.000-07:00
diff --git a/src/sagemaker/serializers.py b/src/sagemaker/serializers.py
@@ -24,7 +24,7 @@
 from sagemaker.utils import DeferredError
 
 try:
-    import scipy
+    import scipy.sparse
 except ImportError as e:
     scipy = DeferredError(e)
 
diff --git a/tests/unit/sagemaker/test_serializers.py b/tests/unit/sagemaker/test_serializers.py
@@ -18,7 +18,7 @@
 
 import numpy as np
 import pytest
-import scipy
+import scipy.sparse
 
 from sagemaker.serializers import (
     CSVSerializer,
