code: def test_lambda_transform_recipe(monkeypatch):
      torch_mock = mock.MagicMock()
      torch_mock.cuda.device_count.return_value = 3
  
      monkeypatch.setattr(functions, "torch", torch_mock)
      monkeypatch.setenv("DATA_OPTIMIZER_GLOBAL_RANK", 2)
  
      called = False
  
      def fn(output_dir, item, device):
          nonlocal called
          assert device == "cuda:2"
          called = True
  
      data_recipe = LambdaDataTransformRecipe(fn, range(1))
  
      data_recipe.prepare_item("", 1)
      assert called
  
imports: import os
  import random
  import sys
  from typing import Any, List
  from unittest import mock
  import lightning_cloud
  import numpy
  import pytest
  import torch
  from lightning import seed_everything
  from lightning.data.streaming import data_processor
  from lightning.data.streaming import functions
  from lightning.data.streaming.cache import Cache, Dir
  from lightning.data.streaming.data_processor import DataChunkRecipe, DataProcessor, DataTransformRecipe, _download_data_target, _get_item_filesizes, _map_items_to_workers_sequentially, _map_items_to_workers_weighted, _remove_target, _upload_fn, _wait_for_disk_usage_higher_than_threshold, _wait_for_file_to_exist
  from lightning.data.streaming.functions import LambdaDataTransformRecipe, map, optimize
  from lightning_cloud import resolver
  from lightning_utilities.core.imports import RequirementCache
label_desc: Use image node_id 20 for calling a global function with example usage: test_lambda_transform_recipe(monkeypatch) without return types