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_gradient_over_time_non_default(self, plt, doctor):
      _, ax = plt.subplots()
      ax_after = doctor.plot_gradient_over_time(
          param_name="lin0.bias",
          ax=ax,
          lw=3,
          bins=np.arange(10),
          color="r",
          figsize=(1, 2),
          interpolate=True,
      )
      assert ax_after is ax
  
imports: import itertools
  from unittest import mock
  import numpy
  import pytest
  import torch
  from torch import nn
label_desc: Use image node_id 45 for calling the TestSkorchDoctorSimple obj's underlying member method code with example usage: obj.test_plot_gradient_over_time_non_default(plt, doctor) without return types