black+isort

9 months ago · 3bffcde0fe
parent 8eb1722f1e
commit 3bffcde0fe
1 changed files with 1 additions and 1 deletions
--- a/src/petals/client/inference_session.py
+++ b/src/petals/client/inference_session.py
@ -255,7 +255,7 @@ class InferenceSession:
                        self._sequence_manager.state.p2p,
                        span,
                        span_uids,
-                        rpc_info=self._sequence_manager.rpc_info, #TODO not actually needed
+                        rpc_info=self._sequence_manager.rpc_info,  # TODO not actually needed
                        max_length=self._max_length,
                        **metadata,
                    )