Function get_serializer

python/pathway/tests/test_py_object_wrapper.py:200–213

(serialization: str)

def get_serializer(serialization: str) -> api.PyObjectWrapperSerializerProtocol | None:
    match serialization:
        case "default":
            return None
        case "pickle":
            return pickle  # type: ignore[return-value]
        case "SimpleStrSerializer":
            return SimpleStrSerializer
        case "dill":
            return dill
        case "cloudpickle":
            return cloudpickle
        case _:
            raise ValueError(f"Incorrect serialization arg: {serialization}.")


@pytest.mark.flaky(reruns=2)