in src/fmeval/model_runners/sm_jumpstart_model_runner.py [0:0]
def __reduce__(self):
"""
Custom serializer method used by Ray when it serializes instances of this
class in eval_algorithms.util.generate_model_predict_response_for_dataset.
"""
serialized_data = (
self._endpoint_name,
self._model_id,
self._content_template,
self._model_version,
self._custom_attributes,
self._output,
self._log_probability,
self._embedding,
self._component_name,
)
return self.__class__, serialized_data