code: def get_expected_model_shape(model_name):
      model_input_shapes = {
          "efficientnet-b0": 224,
          "efficientnet-b1": 240,
          "efficientnet-b2": 260,
          "efficientnet-b3": 300,
          "efficientnet-b4": 380,
          "efficientnet-b5": 456,
          "efficientnet-b6": 528,
          "efficientnet-b7": 600,
      }
      return model_input_shapes[model_name]
  
imports: from __future__ import annotations
  import os
  import unittest
  from typing import TYPE_CHECKING
  from unittest import skipUnless
  import torch
  from parameterized import parameterized
  from monai.networks import eval_mode
  from monai.networks.nets import BlockArgs, EfficientNetBN, EfficientNetBNFeatures, drop_connect, get_efficientnet_image_size
  from monai.utils import optional_import
  from tests.utils import skip_if_downloading_fails, skip_if_quick, test_pretrained_networks, test_script_save
label_desc: Use image node_id 2 for calling a global function with example usage: get_expected_model_shape(model_name) and returns: model_input_shapes