code: def dataset(db):
      return db.get_dataset(1)
  
file_path: ATM/tests/test_core.py
imports: import os
  import pytest
  from atm.core import ATM
  from atm.data import _get_local_path
  from atm.database import Database
label_desc: Use image node_id 2 for calling a global function with example usage: dataset(db) and returns: db