code: def build_submodel(step, number):
      x = Input(name="x_sub{}".format(number))
      y_t = Input(name="y_t_sub{}".format(number))
      y_p = step(x, y_t)
      return Model(x, y_p, y_t)
  
file_path: baikal/tests/test_plot.py
imports: import pytest
  from baikal import Model, Input
  from baikal.plot import plot_model
  from tests.helpers.fixtures import teardown
  from tests.helpers.dummy_steps import DummyMIMO, DummySIMO, DummySISO, DummyMISO
  from tests.helpers.sklearn_steps import LogisticRegression
label_desc: Use image node_id 1 for calling a global function with example usage: build_submodel(step, number) and returns: Model