code: def test_load(tmp_path):
      train_data = create_dataset()
      train(
          "random-forest",
          train_data=train_data,
          test_data=train_data,
          dry_run=True,
          save=tmp_path,
      )
      loaded_model_wrapper = load(tmp_path)
      assert isinstance(loaded_model_wrapper, ModelWrapper)
      assert loaded_model_wrapper.model is not None
  
file_path: traintool/tests/test_main.py
imports: import comet_ml
  import pytest
  from tensorboardX import SummaryWriter
  from conftest import create_dataset
  from traintool.main import connect_comet, train, load, _resolve_model, _write_info_file, _read_info_file, _create_comet_experiment, _create_tensorboard_writer
  from traintool.model_wrapper import ModelWrapper
  from traintool.utils import DummyExperiment
label_desc: Use image node_id 6 for calling a global function with example usage: test_load(tmp_path) without return types