From 3271115acc03f77a283a48a721114821ee37091c Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Fri, 28 Jul 2023 12:11:04 +0800
Subject: [PATCH] small fixes

---
 .github/workflows/build-docker-image.yml |  3 +++
 docker/torch1.12.1-cuda11.3.dockerfile   | 10 +++++++++-
 docker/torch1.13.0-cuda11.6.dockerfile   | 10 +++++++++-
 docker/torch1.9.0-cuda10.2.dockerfile    | 10 +++++++++-
 docker/torch2.0.0-cuda11.7.dockerfile    | 10 +++++++++-
 5 files changed, 39 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/build-docker-image.yml b/.github/workflows/build-docker-image.yml
index 327f0ee45..5d834d087 100644
--- a/.github/workflows/build-docker-image.yml
+++ b/.github/workflows/build-docker-image.yml
@@ -2,6 +2,9 @@
 # https://docs.github.com/en/actions/publishing-packages/publishing-docker-images#publishing-images-to-github-packages
 name: Build docker image
 on:
+  push:
+    branches:
+      - test-yesno-docker
   workflow_dispatch:
 
 concurrency:
diff --git a/docker/torch1.12.1-cuda11.3.dockerfile b/docker/torch1.12.1-cuda11.3.dockerfile
index c5e252abb..48df4405e 100644
--- a/docker/torch1.12.1-cuda11.3.dockerfile
+++ b/docker/torch1.12.1-cuda11.3.dockerfile
@@ -51,7 +51,15 @@ RUN pip install --no-cache-dir \
       sentencepiece>=0.1.96 \
       tensorboard \
       typeguard \
-      dill
+      dill \
+      onnx \
+      onnxruntime \
+      onnxmltools \
+      multi_quantization \
+      typeguard \
+      numpy \
+      pytest \
+      graphviz
 
 RUN git clone https://github.com/k2-fsa/icefall /workspace/icefall && \
     cd /workspace/icefall && \
diff --git a/docker/torch1.13.0-cuda11.6.dockerfile b/docker/torch1.13.0-cuda11.6.dockerfile
index bcbf8b599..4d2f96c8e 100644
--- a/docker/torch1.13.0-cuda11.6.dockerfile
+++ b/docker/torch1.13.0-cuda11.6.dockerfile
@@ -51,7 +51,15 @@ RUN pip install --no-cache-dir \
       sentencepiece>=0.1.96 \
       tensorboard \
       typeguard \
-      dill
+      dill \
+      onnx \
+      onnxruntime \
+      onnxmltools \
+      multi_quantization \
+      typeguard \
+      numpy \
+      pytest \
+      graphviz
 
 RUN git clone https://github.com/k2-fsa/icefall /workspace/icefall && \
     cd /workspace/icefall && \
diff --git a/docker/torch1.9.0-cuda10.2.dockerfile b/docker/torch1.9.0-cuda10.2.dockerfile
index 7553fcf86..9ea45b38a 100644
--- a/docker/torch1.9.0-cuda10.2.dockerfile
+++ b/docker/torch1.9.0-cuda10.2.dockerfile
@@ -51,7 +51,15 @@ RUN pip install --no-cache-dir \
       sentencepiece>=0.1.96 \
       tensorboard \
       typeguard \
-      dill
+      dill \
+      onnx \
+      onnxruntime \
+      onnxmltools \
+      multi_quantization \
+      typeguard \
+      numpy \
+      pytest \
+      graphviz
 
 RUN git clone https://github.com/k2-fsa/icefall /workspace/icefall && \
     cd /workspace/icefall && \
diff --git a/docker/torch2.0.0-cuda11.7.dockerfile b/docker/torch2.0.0-cuda11.7.dockerfile
index c11c0bd67..9dd4d51c9 100644
--- a/docker/torch2.0.0-cuda11.7.dockerfile
+++ b/docker/torch2.0.0-cuda11.7.dockerfile
@@ -51,7 +51,15 @@ RUN pip install --no-cache-dir \
       sentencepiece>=0.1.96 \
       tensorboard \
       typeguard \
-      dill
+      dill \
+      onnx \
+      onnxruntime \
+      onnxmltools \
+      multi_quantization \
+      typeguard \
+      numpy \
+      pytest \
+      graphviz
 
 RUN git clone https://github.com/k2-fsa/icefall /workspace/icefall && \
     cd /workspace/icefall && \