diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/__init__.py b/egs/librispeech/ASR/pruned_transducer_stateless4/__init__.py
similarity index 100%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/__init__.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/__init__.py
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/asr_datamodule.py b/egs/librispeech/ASR/pruned_transducer_stateless4/asr_datamodule.py
similarity index 100%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/asr_datamodule.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/asr_datamodule.py
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/beam_search.py b/egs/librispeech/ASR/pruned_transducer_stateless4/beam_search.py
similarity index 100%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/beam_search.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/beam_search.py
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless4/conformer.py
similarity index 100%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/conformer.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/conformer.py
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/decode.py b/egs/librispeech/ASR/pruned_transducer_stateless4/decode.py
similarity index 100%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/decode.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/decode.py
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/decoder.py b/egs/librispeech/ASR/pruned_transducer_stateless4/decoder.py
similarity index 100%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/decoder.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/decoder.py
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/encoder_interface.py b/egs/librispeech/ASR/pruned_transducer_stateless4/encoder_interface.py
similarity index 100%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/encoder_interface.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/encoder_interface.py
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/export.py b/egs/librispeech/ASR/pruned_transducer_stateless4/export.py
similarity index 100%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/export.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/export.py
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/joiner.py b/egs/librispeech/ASR/pruned_transducer_stateless4/joiner.py
similarity index 100%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/joiner.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/joiner.py
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/model.py b/egs/librispeech/ASR/pruned_transducer_stateless4/model.py
similarity index 100%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/model.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/model.py
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/optim.py b/egs/librispeech/ASR/pruned_transducer_stateless4/optim.py
similarity index 100%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/optim.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/optim.py
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/scaling.py b/egs/librispeech/ASR/pruned_transducer_stateless4/scaling.py
similarity index 100%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/scaling.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/scaling.py
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/train.py b/egs/librispeech/ASR/pruned_transducer_stateless4/train.py
similarity index 99%
rename from egs/librispeech/ASR/pruned_transducer_stateless3/train.py
rename to egs/librispeech/ASR/pruned_transducer_stateless4/train.py
index 83d31c64e..33478c630 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless3/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless4/train.py
@@ -74,8 +74,10 @@ from torch.utils.tensorboard import SummaryWriter
 from icefall import diagnostics
 from icefall.checkpoint import load_checkpoint, remove_checkpoints
 from icefall.checkpoint import save_checkpoint as save_checkpoint_impl
-from icefall.checkpoint import save_checkpoint_with_global_batch_idx
-from icefall.checkpoint import update_averaged_model
+from icefall.checkpoint import (
+    save_checkpoint_with_global_batch_idx,
+    update_averaged_model,
+)
 from icefall.dist import cleanup_dist, setup_dist
 from icefall.env import get_env_info
 from icefall.utils import AttributeDict, MetricsTracker, setup_logger, str2bool