Skip to content

Commit 52251be

Browse files
fix: fix lint error
1 parent 2af78f9 commit 52251be

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

graphgen/models/llm/local/vllm_wrapper.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -113,18 +113,17 @@ async def generate_topk_per_token(
113113
tok_str = logprob_obj.decoded_token.strip() if logprob_obj.decoded_token else ""
114114
prob = float(math.exp(logprob_obj.logprob))
115115
candidate_tokens.append(Token(tok_str, prob))
116-
116+
117117
candidate_tokens.sort(key=lambda x: -x.prob)
118-
118+
119119
if candidate_tokens:
120120
main_token = Token(
121121
text=candidate_tokens[0].text,
122122
prob=candidate_tokens[0].prob,
123123
top_candidates=candidate_tokens
124124
)
125125
return [main_token]
126-
else:
127-
return []
126+
return []
128127

129128
async def generate_inputs_prob(
130129
self, text: str, history: Optional[List[str]] = None, **extra: Any

0 commit comments

Comments
 (0)