code: def gpu_count(device):
      if device == "cpu":
          return 0
      else:
          return 1
  
file_path: collie/tests/conftest.py
imports: import sys
  import os
  import warnings
  from .fixtures.cross_validation_fixtures import *
  from .fixtures.interactions_fixtures import *
  from .fixtures.loss_fixtures import *
  from .fixtures.metrics_fixtures import *
  from .fixtures.model_fixtures import *
  from .fixtures.movielens_fixtures import *
  from .fixtures.utils_fixtures import *
label_desc: Use image node_id 2 for calling a global function with example usage: gpu_count(device) and returns: int, int