Method sub

verl/tests/single_controller/test_colocated_workers.py:47–49

(self, data: DataProto)

    @register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)
    async def sub(self, data: DataProto):
        data.batch["a"] -= self.config["b"]
        return data


def test_colocated_workers():