code: def test_polynomial_decay_initialization_failure(steps):
      with pytest.raises(Exception):
          PolynomialDecayScheduler(update_steps=steps)
      with pytest.raises(Exception):
          PolynomialDecayScheduler(update_steps=torch.tensor(steps))
  
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 3 for calling a global function with example usage: test_polynomial_decay_initialization_failure(steps) without return types