code: def mock_sampler(input_iter):
      return input_iter
  
file_path: typedb-ml/typedb_ml/networkx/queries_to_networkx_it.py
imports: import sys
  import unittest
  import networkx
  from typedb.client import *
  from typedb_ml.networkx.graph_test_case import GraphTestCase
  from typedb_ml.networkx.queries_to_networkx import build_graph_from_queries
  from typedb_ml.networkx.query_graph import Query
  from typedb_ml.typedb.test.base import TypeDBServer
  from typedb_ml.typedb.thing import build_thing
label_desc: Use image node_id 1 for calling a global function with example usage: mock_sampler(input_iter) and returns: input_iter