code: def dict_of_ndarrays_with_nans():
      return {
          "1D": np.array([0.5, np.nan, 2.0]),
          "2D": np.array([[0.1, 0.2], [np.nan, 0.5]]),
          "3D": np.array(
              [
                  [[0.1, np.nan], [0.3, 0.4]],
                  [[np.nan, 0.6], [0.7, np.nan]],
              ]
          ),
      }
  
imports: import json
  import math
  from unittest import mock
  import numpy
  import pandas
  import pytest
  import sklearn.neighbors
  from scipy.sparse import csc_matrix, csr_matrix
  from sklearn import datasets
  from sklearn.base import BaseEstimator, ClassifierMixin
  import mlflow
  from mlflow.models import Model
  from mlflow.models.signature import ModelSignature, infer_signature
  from mlflow.models.utils import _Example, _read_sparse_matrix_from_json, _read_tensor_input_from_json
  from mlflow.types import DataType
  from mlflow.types.schema import ColSpec, Schema, TensorSpec
  from mlflow.types.utils import TensorsNotSupportedException
  from mlflow.utils.file_utils import TempDir
  from mlflow.utils.proto_json_utils import dataframe_from_raw_json
  from tests.helper_functions import AnyStringWith
label_desc: Use image node_id 5 for calling a global function with example usage: dict_of_ndarrays_with_nans() and returns: dict