From 09914053619fe78232ec39931c39947b35717f9e Mon Sep 17 00:00:00 2001 From: yuxuan-lou <83441848+yuxuan-lou@users.noreply.github.com> Date: Wed, 19 Jul 2023 22:18:30 +0800 Subject: [PATCH] [NFC] polish applications/Chat/coati/models/utils.py codestyle (#4277) * [NFC] polish colossalai/context/random/__init__.py code style * [NFC] polish applications/Chat/coati/models/utils.py code style --- applications/Chat/coati/models/utils.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/applications/Chat/coati/models/utils.py b/applications/Chat/coati/models/utils.py index b9f15f894..772bfc329 100644 --- a/applications/Chat/coati/models/utils.py +++ b/applications/Chat/coati/models/utils.py @@ -46,10 +46,7 @@ def log_probs_from_logits(logits: torch.Tensor, labels: torch.Tensor) -> torch.T return log_probs_labels.squeeze(-1) -def calc_action_log_probs(output: torch.Tensor, - sequences: torch.LongTensor, - num_actions: int - ) -> torch.Tensor: +def calc_action_log_probs(output: torch.Tensor, sequences: torch.LongTensor, num_actions: int) -> torch.Tensor: """Calculate action log probs. Args: