Skip to content

Commit c4dbe37

Browse files
authored
Fix lint failures on main (#110)
Signed-off-by: zhyncs <46627482+zhyncs@users.noreply.github.com>
1 parent 20c2f9c commit c4dbe37

2 files changed

Lines changed: 2 additions & 4 deletions

File tree

torchspec/inference/engine/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
]
2929

3030
try:
31-
from torchspec.inference.engine.hf_runner import HFRunner
31+
from torchspec.inference.engine.hf_runner import HFRunner # noqa: F401
3232

3333
__all__.append("HFRunner")
3434
except ImportError as _e:

torchspec/inference/engine/sgl_engine_decode.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -211,9 +211,7 @@ def generate_with_decode(
211211
top_k = getattr(self.args, "decode_top_k", -1)
212212
if top_k > 0:
213213
sampling_params["top_k"] = top_k
214-
logger.debug(
215-
f"SglEngine rank {self.rank}: decode sampling_params={sampling_params}"
216-
)
214+
logger.debug(f"SglEngine rank {self.rank}: decode sampling_params={sampling_params}")
217215

218216
outputs: list[dict[str, Any] | None] = [None for _ in range(batch_size)]
219217
active_indices = list(range(batch_size))

0 commit comments

Comments
 (0)