code: def test_ckpt_non_tensor_input(self, non_tensor):
      module = LinearNonTensorInput(HIDDEN_DIM, HIDDEN_DIM)
      inputs = torch.rand(HIDDEN_DIM)
      inputs.requires_grad = True
      _test_activation_checkpoint(module, inputs, non_tensor)
  
imports: import pytest
  import torch
  import deepspeed
  from copy import deepcopy
  from unit.common import DistributedTest
label_desc: Use image node_id 1 for calling the TestCheckpointNonTensor obj's underlying member method code with example usage: obj.test_ckpt_non_tensor_input(non_tensor) without return types