code: def test_wrapper_initialization(
      max_ragged_sequence_count: int, max_ragged_batch_size: int
  ) -> None:
      config = DSStateManagerConfig(
          max_tracked_sequences=max_ragged_sequence_count,
          max_ragged_batch_size=max_ragged_batch_size,
          max_ragged_sequence_count=max_ragged_sequence_count,
      )
  
      batch = RaggedBatchWrapper(config)
  
      assert batch.current_tokens == 0
      assert batch.current_sequences == 0
  
file_path: deepspeed/tests/unit/inference/v2/ragged/test_ragged_wrapper.py
imports: from typing import List
  import pytest
  import torch
  from deepspeed.accelerator import get_accelerator
  from deepspeed.inference.v2.ragged import PlaceholderSequenceDescriptor, RaggedBatchWrapper, DSStateManagerConfig
label_desc: Use image node_id 1 for calling a global function with example usage: test_wrapper_initialization(max_ragged_sequence_count, max_ragged_batch_size) without return types