code: def __init__(self, mock_data, driver_version="535.104.12"):
      self._mock_data = mock_data
      self.driver_version = driver_version
  
imports: import pytest
  from typing import List
  from unittest.mock import patch
  import ray._private.thirdparty.pynvml
label_desc: Use image node_id 1 to create a new PyNVMLMock object  with example: obj = PyNVMLMock(mock_data, driver_version)