Method update

tasks/AutoMolecule3D/code/experiment.py:305–306

(self, inputs: Tuple[torch.Tensor, torch.Tensor])

        return x, vec

    def update(self, inputs: Tuple[torch.Tensor, torch.Tensor]) -> Tuple[torch.Tensor, torch.Tensor]:
        return inputs
    
class ViS_MP_Vertex_Edge(ViS_MP):