Skip to content

Commit 59bd98b

Browse files
committed
Fix new issues
1 parent 7279adf commit 59bd98b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/mlmodel_gemini/_mock_external_gemini_server.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -912,7 +912,7 @@ def _simple_get(self):
912912
if not prompt:
913913
self.send_response(500)
914914
self.end_headers()
915-
self.wfile.write("Could not parse prompt.".encode("utf-8"))
915+
self.wfile.write(b"Could not parse prompt.")
916916
return
917917

918918
headers, response = ({}, "")
@@ -924,7 +924,7 @@ def _simple_get(self):
924924
else: # If no matches found
925925
self.send_response(500)
926926
self.end_headers()
927-
self.wfile.write(("Unknown Prompt:\n%s" % prompt).encode("utf-8"))
927+
self.wfile.write(f"Unknown Prompt:\n{prompt}".encode())
928928
return
929929

930930
# Send response code
@@ -967,13 +967,13 @@ class _MockExternalGeminiServer(MockExternalHTTPServer):
967967
# calls.
968968

969969
def __init__(self, handler=simple_get, port=None, *args, **kwargs):
970-
super(_MockExternalGeminiServer, self).__init__(handler=handler, port=port, *args, **kwargs) # noqa: B026
970+
super().__init__(handler=handler, port=port, *args, **kwargs) # noqa: B026
971971

972972
return _MockExternalGeminiServer
973973

974974

975975
if __name__ == "__main__":
976976
with MockExternalGeminiServer() as server:
977-
print("MockExternalGeminiServer serving on port %s" % str(server.port))
977+
print(f"MockExternalGeminiServer serving on port {str(server.port)}")
978978
while True:
979979
pass # Serve forever

0 commit comments

Comments
 (0)