From 7f4749a4d6d5443be77035e9c164c1b8fe9ac40c Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Wed, 22 Dec 2021 15:26:26 +0800
Subject: [PATCH] Fix CI errors.

---
 .github/workflows/run-pretrained.yml               |  2 +-
 .github/workflows/test.yml                         | 14 +++++++++++++-
 egs/librispeech/ASR/README.md                      |  3 +--
 .../ASR/transducer_stateless/test_decoder.py       |  2 --
 4 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/run-pretrained.yml b/.github/workflows/run-pretrained.yml
index 710ca2603..1758a3521 100644
--- a/.github/workflows/run-pretrained.yml
+++ b/.github/workflows/run-pretrained.yml
@@ -30,7 +30,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-18.04]
-        python-version: [3.6, 3.7, 3.8, 3.9]
+        python-version: [3.7, 3.8, 3.9]
         torch: ["1.10.0"]
         torchaudio: ["0.10.0"]
         k2-version: ["1.9.dev20211101"]
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index baa9c1727..f2c63a3b8 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -32,7 +32,7 @@ jobs:
         # os: [ubuntu-18.04, macos-10.15]
         # disable macOS test for now.
         os: [ubuntu-18.04]
-        python-version: [3.6, 3.7, 3.8, 3.9]
+        python-version: [3.7, 3.8]
         torch: ["1.8.0", "1.10.0"]
         torchaudio: ["0.8.0", "0.10.0"]
         k2-version: ["1.9.dev20211101"]
@@ -106,6 +106,12 @@ jobs:
           if [[ ${{ matrix.torchaudio }} == "0.10.0" ]]; then
             cd ../transducer
             pytest -v -s
+
+            cd ../transducer_stateless
+            pytest -v -s
+
+            cd ../transducer_lstm
+            pytest -v -s
           fi
 
       - name: Run tests
@@ -125,4 +131,10 @@ jobs:
           if [[ ${{ matrix.torchaudio }} == "0.10.0" ]]; then
             cd ../transducer
             pytest -v -s
+
+            cd ../transducer_stateless
+            pytest -v -s
+
+            cd ../transducer_lstm
+            pytest -v -s
           fi
diff --git a/egs/librispeech/ASR/README.md b/egs/librispeech/ASR/README.md
index 76113eff7..1ee12b0ab 100644
--- a/egs/librispeech/ASR/README.md
+++ b/egs/librispeech/ASR/README.md
@@ -13,5 +13,4 @@ The following table lists the differences among them.
 |------------------------|-----------|--------------------|
 | `transducer`           | Conformer | LSTM               |
 | `transducer_stateless` | Conformer | Conv1d + Embedding |
-
-
+| `transducer_lstm     ` | LSTM      | LSTM               |
diff --git a/egs/librispeech/ASR/transducer_stateless/test_decoder.py b/egs/librispeech/ASR/transducer_stateless/test_decoder.py
index fa6632eb7..3a653c1b7 100755
--- a/egs/librispeech/ASR/transducer_stateless/test_decoder.py
+++ b/egs/librispeech/ASR/transducer_stateless/test_decoder.py
@@ -23,8 +23,6 @@ To run this file, do:
 """
 
 import torch
-import torch.nn as nn
-import torch.nn.functional as F
 from decoder import Decoder