switching backend to agg on matplotlib package install

Zero-True · Jan 9, 2025 · 29dc2a7 · 29dc2a7
1 parent 9202e17
commit 29dc2a7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/zt_backend/utils/dependencies.py b/zt_backend/utils/dependencies.py
@@ -73,6 +73,12 @@ async def dependency_update(
                 await websocket.send_json({"output": line})
             process.stdout.close()
         write_dependencies(dependency_request.dependencies)
+        if any(dep.startswith("matplotlib") for dep in dependency_request.dependencies):
+            try:
+                import matplotlib
+                matplotlib.use("Agg")
+            except Exception as e:
+                logger.info('matplotlib not found')
         return parse_dependencies()
     except Exception as e:
         logger.error("Error updating dependencies: %s", traceback.format_exc())