[misc] add zmq in collect env (#7119)

2024-08-03 22:03:46 -07:00
parent 654bc5ca49
commit 9fadc7b7a0
1 changed files with 2 additions and 0 deletions
--- a/collect_env.py
+++ b/collect_env.py
@@ -65,6 +65,7 @@ DEFAULT_CONDA_PATTERNS = {
    "optree",
    "nccl",
    "transformers",
+    "zmq",
 }

 DEFAULT_PIP_PATTERNS = {
@@ -77,6 +78,7 @@ DEFAULT_PIP_PATTERNS = {
    "onnx",
    "nccl",
    "transformers",
+    "zmq",
 }