From 952abbd5d979d081c5923a2cc1e75ca09e77e0ca Mon Sep 17 00:00:00 2001 From: robby-wang Date: Wed, 17 Jan 2024 19:46:31 -0500 Subject: [PATCH] Changed docstring of readout return type: float -> torch.Tensor --- mtenn/readout.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mtenn/readout.py b/mtenn/readout.py index 0110c48..cebf05c 100644 --- a/mtenn/readout.py +++ b/mtenn/readout.py @@ -70,7 +70,7 @@ def forward(self, delta_g): Returns ------- - float + torch.Tensor Calculated pIC50 value. """ pic50 = -delta_g / torch.log(torch.tensor(10, dtype=delta_g.dtype)) @@ -116,7 +116,7 @@ def forward(self, delta_g): Returns ------- - float + torch.Tensor Calculated Ki value. """ ki = torch.exp(delta_g)