code: def test_prediction_error(monkeypatch, model):
      monkeypatch.setenv("MODELKIT_ENABLE_SIMPLE_TRACEBACK", True)
      with pytest.raises(CustomError) as excinfo:
          model.predict({})
      assert len(excinfo.traceback) <= 3
  
      with pytest.raises(CustomError) as excinfo:
          model.predict_batch([{}])
      assert len(excinfo.traceback) <= 3
  
      with pytest.raises(CustomError) as excinfo:
          next(model.predict_gen(iter(({},))))
      assert len(excinfo.traceback) <= 3
  
imports: import pytest
  from modelkit.core.model import AsyncModel, Model
label_desc: Use image node_id 1 for calling a global function with example usage: test_prediction_error(monkeypatch, model) without return types