Complete forgotten await

Author: ayjayt

src/py/kaleido/kaleido.py

@@ -34,10 +34,10 @@
 _stepper = False
 
 
-def _to_thread(func, *args, **kwargs):
+async def _to_thread(func, *args, **kwargs):
     _loop = asyncio.get_running_loop()
     fn = partial(func, *args, **kwargs)
-    _loop.run_in_executor(None, fn)
+    await _loop.run_in_executor(None, fn)
 
 
 # this is kinda public but undocumented
@@ -376,7 +376,7 @@ def write_image(binary):
                 file.write(binary)
 
         _logger.info(f"Starting write of {full_path.name}")
-        _to_thread(write_image, img)
+        await _to_thread(write_image, img)
         _logger.info(f"Wrote {full_path.name}")
         if profiler is not None:
             self._finish_profile(profile, e, full_path.stat().st_size / 1000000)