code: def f2(p, t):
      """Compute the function p * cos(t**2)."""
      return np.cos(p * t**2)
  
file_path: pennylane/tests/pulse/test_parametrized_hamiltonian_pytree.py
imports: import numpy
  import pytest
  import pennylane
label_desc: Use image node_id 2 for calling a global function with example usage: f2(p, t) and returns: np