code: def only_register(
      callback_fn, module, overwrite
  ):  # pylint: disable=unused-argument
      mlflow.utils.import_hooks._post_import_hooks[module] = [
          callback_fn
      ]
  
imports: import sys
  from collections import namedtuple
  from io import StringIO
  from unittest import mock
  import fastai
  import lightgbm
  import pyspark
  import pyspark.ml
  import pytest
  import pytorch_lightning
  import setfit
  import sklearn
  import statsmodels
  import tensorflow
  import transformers
  import xgboost
  import mlflow
  from mlflow.utils.autologging_utils import AutologgingEventLogger, autologging_is_disabled, get_autologging_config
  from tests.autologging.fixtures import reset_stderr, test_mode_off, test_mode_on
label_desc: Use image node_id 2 for calling a global function with example usage: only_register(callback_fn, module, overwrite) without return types