Enable some existing integ tests

Author: Ao Guo

requirements/extras/test_requirements.txt

@@ -22,4 +22,3 @@ Jinja2==3.0.3
 pandas>=1.3.5,<1.5
 scikit-learn==1.0.2
 cloudpickle==2.2.0
-tblib==1.7.0

tests/integ/sagemaker/remote_function/test_decorator.py

@@ -17,10 +17,10 @@
 import os
 
 from sagemaker.remote_function import remote
+from sagemaker.remote_function.errors import RuntimeEnvironmentError
 
 from tests.integ.kms_utils import get_or_create_kms_key
 from tests.integ import DATA_DIR
-from sagemaker.exceptions import UnexpectedStatusException
 
 ROLE = "SageMakerRole"
 
@@ -45,7 +45,6 @@ def divide(x, y):
     assert divide(20, 2) == 10
 
 
-@pytest.mark.skip
 def test_decorated_function_raises_exception(
     sagemaker_session, dummy_container_without_error, cpu_instance_type
 ):
@@ -133,11 +132,7 @@ def test_with_non_existent_dependencies(
     def divide(x, y):
         return x / y
 
-    # TODO: this should raise RuntimeEnvironmentError
-    # Currently this is returning UnexpectedStatusException due to how `job.wait()` is implemented
-    # in remote_function's client. We are correctly throwing  RuntimeEnvironmentError from
-    # Runtime env module
-    with pytest.raises(UnexpectedStatusException):
+    with pytest.raises(RuntimeEnvironmentError):
         divide(10, 2)