code: def simple_module():
      return torch.nn.Conv2d(3, 3, (3, 3), bias=False, groups=1)
  
file_path: coremltools/coremltools/test/optimize/torch/pruning/test_pruning_scheduler.py
imports: import sys
  import pytest
  import torch
  from coremltools.optimize.torch.pruning import ConstantSparsityScheduler, MagnitudePruner, MagnitudePrunerConfig, ModuleMagnitudePrunerConfig, PolynomialDecayScheduler
label_desc: Use image node_id 1 for calling a global function with example usage: simple_module() and returns: torch