chore: import upstream snapshot with attribution

python/ray/air/tests/conftest.py

@@ -0,0 +1,23 @@
+# Trigger pytest hook to automatically zip test cluster logs to archive dir on failure
+import copy
+
+import pytest
+
+import ray
+from ray.tests.conftest import pytest_runtest_makereport  # noqa
+
+
+@pytest.fixture
+def restore_data_context(request):
+    """Restore any DataContext changes after the test runs"""
+    original = copy.deepcopy(ray.data.context.DataContext.get_current())
+    yield
+    ray.data.context.DataContext._set_current(original)
+
+
+@pytest.fixture
+def disable_fallback_to_object_extension(request, restore_data_context):
+    """Disables fallback to ArrowPythonObjectType"""
+    ray.data.context.DataContext.get_current().enable_fallback_to_arrow_object_ext_type = (
+        False
+    )