class_docstr: Test functionality of SkorchDoctor using a simple model
class_docstr_tok: ["Test","functionality","of","SkorchDoctor","using","a","simple","model"]
code: def test_plot_param_updates_non_default(self, plt, doctor):
      _, axes = plt.subplots(1, 1, squeeze=False)
      axes_after = doctor.plot_param_updates(
          axes=axes,
          match_fn=lambda name: "lin" in name,
          lw=3,
          figsize=(1, 2),
      )
      assert isinstance(axes, np.ndarray)
      assert axes_after is axes
  
imports: import itertools
  from unittest import mock
  import numpy
  import pytest
  import torch
  from torch import nn
label_desc: Use image node_id 39 for calling the TestSkorchDoctorSimple obj's underlying member method code with example usage: obj.test_plot_param_updates_non_default(plt, doctor) without return types