docstr: Assert that non-rank-0 workers get a noop module
docstr_tok: ["Assert","that","non-rank-0","workers","get","a","noop","module"]
code: def testMlFlowSetupRankNonRankZero(self):
      """Assert that non-rank-0 workers get a noop module"""
      storage = StorageContext(
          storage_path=tempfile.mkdtemp(),
          experiment_dir_name="exp_name",
          trial_dir_name="trial_name",
      )
  
      init_session(
          training_func=None,
          world_rank=1,
          local_rank=1,
          node_rank=1,
          local_world_size=2,
          world_size=2,
          storage=storage,
      )
      mlflow = setup_mlflow({})
      assert isinstance(mlflow, _NoopModule)
  
      mlflow.log_metrics()
      mlflow.sklearn.save_model(None, "model_directory")
  
imports: import os
  import shutil
  import tempfile
  import unittest
  from collections import namedtuple
  from unittest.mock import patch
  from mlflow.tracking import MlflowClient
  from ray._private.dict import flatten_dict
  from ray.train._internal.session import init_session, shutdown_session
  from ray.train._internal.storage import StorageContext
  from ray.air.integrations.mlflow import MLflowLoggerCallback, setup_mlflow, _NoopModule
  from ray.air._internal.mlflow import _MLflowLoggerUtil
label_desc: Use image node_id 6 for calling the MLflowTest obj's underlying member method code with example usage: obj.testMlFlowSetupRankNonRankZero() without return types