fix incorrect strutlog usage throwing an error when setup fails

replicate · Jul 26, 2024 · bb6f901 · bb6f901
1 parent 9f49b29
commit bb6f901
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/cog/server/connection.py b/python/cog/server/connection.py
@@ -23,7 +23,7 @@ async def async_init(self) -> None:
         # mp may have handled something already but let's dup so exit is clean
         dup_fd = os.dup(fd)
         sock = socket.socket(fileno=dup_fd)
-        sock.setblocking(False)
+        sock.setblocking(True)
         # TODO: use /proc/sys/net/core/rmem_max, but special-case language models
         sz = 65536
         sock.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, sz)

diff --git a/python/cog/server/runner.py b/python/cog/server/runner.py
@@ -199,7 +199,7 @@ async def inner() -> SetupResult:
                 logs.append(traceback.format_exc())
                 status = schema.Status.FAILED
             except BaseException:
-                self.log("caught BaseException during setup, did something go wrong?")
+                self.log.info("caught BaseException during setup, did something go wrong?")
                 logs.append(traceback.format_exc())
                 status = schema.Status.FAILED