code: def check_encoder_on_device(self, device):
      priming_data = generate_sentences(2, 6, vocab_size=99)
      enc = ShortTextEncoder(is_target=True, device=device)
      enc.prepare(priming_data)
      self.assertEqual(
          list(enc.cae.net.parameters())[0].device.type, device
      )
  
file_path: lightwood/tests/unit_tests/encoder/text/test_short.py
imports: import random
  import unittest
  from lightwood.encoder.text.short import ShortTextEncoder
  from type_infer.rule_based.helpers import tokenize_text
  import torch
label_desc: Use image node_id 7 for calling the TestShortTextEncoder obj's underlying member method code with example usage: obj.check_encoder_on_device(device) without return types