code: def fig_make(m_rows, n_cols, flatten, **kwargs):
      """Gets figure for plotting with m x n axes."""
      figsize = plt.rcParams["figure.figsize"]
      figsize = (figsize[0] * n_cols, figsize[1] * m_rows)
      fig, axes = plt.subplots(
          m_rows, n_cols, figsize=figsize, squeeze=False, **kwargs
      )
      axes = [ax for axes in axes for ax in axes] if flatten else axes
      return fig, axes
  
file_path: pycls/pycls/sweep/plotting.py
imports: import matplotlib.lines
  import matplotlib.pyplot
  import matplotlib.ticker
  import numpy
  import pycls.models.regnet
  from pycls.sweep.analysis import get_info, get_vals, sort_sweep
label_desc: Use image node_id 5 for calling a global function with example usage: fig_make(m_rows, n_cols, flatten) and returns: fig,  axes