2708ccddd59bc7be358dc8ae8297dce26f545741,grid/websocket_client.py,WebsocketGridClient,run_remote_inference,#WebsocketGridClient#Any#Any#Any#,382

Before Change

            )
        )

        return self._return_bool_result(result, return_key="prediction")

After Change

            "encoding": self._encoding,
        }
        self.ws.send(json.dumps(payload))
        response = json.loads(self.ws.recv())
        if response["success"]:
            return torch.tensor(response["prediction"])
        else:
            raise RuntimeError(response["error"])

Instances