code: def test_update_tensor_devices(mode):
      model = HiFiCAutoencoder()
  
      model.update_tensor_devices(mode)
  
file_path: NeuralCompression/tests/models/test_hific_autoencoder.py
imports: import pytest
  import torch
  from torch import Tensor
  from neuralcompression.models import HiFiCAutoencoder
label_desc: Use image node_id 4 for calling a global function with example usage: test_update_tensor_devices(mode) without return types