From a59f593f6f904d245c32be0c445fcdf1d607fac6 Mon Sep 17 00:00:00 2001 From: RaulPPealez Date: Fri, 3 Nov 2023 13:23:39 +0100 Subject: [PATCH] Remove reduce --- torchmdnet/models/model.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/torchmdnet/models/model.py b/torchmdnet/models/model.py index 181266a85..e80c6b93d 100644 --- a/torchmdnet/models/model.py +++ b/torchmdnet/models/model.py @@ -313,9 +313,6 @@ def reset_parameters(self): def per_point(self, point_features, results, z, pos, batch, extra_args): return point_features, results - def reduce(self, point_features, results, z, pos, batch, extra_args): - return point_features, results - def per_sample(self, point_features, results, z, pos, batch, extra_args): return point_features, results