Fixing failing tests and mypy check

Author: svartalf

sentry_sdk/integrations/ray.py

@@ -14,7 +14,7 @@
 )
 
 try:
-    import ray  # type: ignore[import-not-found]
+    import ray
 except ImportError:
     raise DidNotEnable("Ray not installed.")
 import functools
@@ -54,7 +54,6 @@ def new_remote(f=None, *args, **kwargs):
 
         def wrapper(user_f):
             # type: (Callable[..., Any]) -> Any
-            @functools.wraps(user_f)
             def new_func(*f_args, _tracing=None, **f_kwargs):
                 # type: (Any, Optional[dict[str, Any]], Any) -> Any
                 _check_sentry_initialized()

tests/integrations/ray/test_ray.py

@@ -59,7 +59,9 @@ def read_error_from_log(job_id):
 
 
 @pytest.mark.forked
-@pytest.mark.parametrize("task_options", [{}, {"num_cpus": 0}])
+@pytest.mark.parametrize(
+    "task_options", [{}, {"num_cpus": 0, "memory": 1024 * 1024 * 10}]
+)
 def test_tracing_in_ray_tasks(task_options):
     setup_sentry()