76f937a915bff2383b830cd63c9b38d32f9fdee8,test/test_api_grid.py,GridAPITest,test_grid_search,#GridAPITest#,27

Before Change

        self.assertTrue("James" in response["grid-nodes"])

    def test_grid_search(self):
        nodes = []
        for (node_id, port) in zip(IDS, PORTS):
            node = gr.WebsocketGridClient(
                hook, "http://localhost:" + port + "/", id=node_id
            )
            node.connect()
            nodes.append(node)

        x = th.tensor([1, 2, 3, 4, 5]).tag("//simple-tensor").describe("Simple tensor")
        y = (
            th.tensor([[4], [5], [7], [8]])
            .tag("/ǘd-tensor")
            .describe("2d tensor example")
        )
        z = (
            th.tensor([[0, 0, 0, 0, 0]])
            .tag("//zeros-tensor")
            .describe("tensor with zeros")
        )
        w = (
            th.tensor([[0, 0, 0, 0, 0]])
            .tag("//zeros-tensor")
            .describe("tensor with zeros")
        )

        x_s = x.send(nodes[0])
        y_s = y.send(nodes[1])
        z_s = z.send(nodes[2])
        w_s = w.send(nodes[0])

        x_s.child.garbage_collect_data = False

After Change

        self.assertTrue("james" in response["grid-nodes"])

    def test_grid_search(self):
        nodes = self.connect_nodes()
        alice, bob, james = nodes["alice"], nodes["bob"], nodes["james"]

        simple_tensor = (
            th.tensor([1, 2, 3, 4, 5]).tag("//simple-tensor").describe("Simple tensor")

Instances