code: def test_arrowds():
      dataset = HFdata.load_dataset("glue", "sst2")["test"]
      dataset = ActiveLearningDataset(dataset)
      dataset.label(np.arange(10))
      assert len(dataset) == 10
      assert len(dataset.pool) == 1811
      data = dataset.pool[0]
      assert [
          k in ["idx", "label", "sentence"] for k, v in data.items()
      ]
  
imports: import pickle
  import unittest
  import unittest.mock
  import warnings
  import datasets
  import numpy
  import pytest
  import torch.utils.data
  from torchvision.transforms import Lambda
  from baal.active import ActiveLearningDataset
label_desc: Use image node_id 1 for calling a global function with example usage: test_arrowds() without return types