From 374751427a90bd96878e9d3e59b5aae5be4f6455 Mon Sep 17 00:00:00 2001
From: Siyuan Tian <siyuant@vmware.com>
Date: Mon, 14 Aug 2023 12:32:24 +0800
Subject: [PATCH] add missing module six and change deprecated data type from
 np.float to np.float64

---
 .../sequence_parallel/data/datasets/indexed_dataset.py        | 4 ++--
 examples/tutorial/sequence_parallel/requirements.txt          | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/examples/tutorial/sequence_parallel/data/datasets/indexed_dataset.py b/examples/tutorial/sequence_parallel/data/datasets/indexed_dataset.py
index b4febcd822e1..e605f608054c 100644
--- a/examples/tutorial/sequence_parallel/data/datasets/indexed_dataset.py
+++ b/examples/tutorial/sequence_parallel/data/datasets/indexed_dataset.py
@@ -94,7 +94,7 @@ def write_longs(f, a):
     3: np.int16,
     4: np.int32,
     5: np.int64,
-    6: np.float,
+    6: np.float64,
     7: np.double,
     8: np.uint16
 }
@@ -267,7 +267,7 @@ class IndexedDatasetBuilder(object):
         np.int16: 2,
         np.int32: 4,
         np.int64: 8,
-        np.float: 4,
+        np.float64: 4,
         np.double: 8
     }
 
diff --git a/examples/tutorial/sequence_parallel/requirements.txt b/examples/tutorial/sequence_parallel/requirements.txt
index b49a94554afb..4fc576453de8 100644
--- a/examples/tutorial/sequence_parallel/requirements.txt
+++ b/examples/tutorial/sequence_parallel/requirements.txt
@@ -1,2 +1,3 @@
 colossalai
 torch
+six