core.units.mlip_unit.inference.client_websocket

Contents

core.units.mlip_unit.inference.client_websocket#

Classes#

`AsyncMLIPInferenceWebSocketClient`
`SyncMLIPInferenceWebSocketClient`

Module Contents#

class core.units.mlip_unit.inference.client_websocket.AsyncMLIPInferenceWebSocketClient(host, port, max_size=100 * 1024 * 1024)#

uri#

websocket = None#

max_size#

async connect()#

async close()#

async call(atomic_data)#

class core.units.mlip_unit.inference.client_websocket.SyncMLIPInferenceWebSocketClient(host, port, max_size=100 * 1024 * 1024)#

uri#

ws = None#

max_size#

connect()#

call(atomic_data)#

__del__()#

close()#