add missing module six and change deprecated data type from np.float …

…to np.float64
hpcaitech · Aug 14, 2023 · 3747514 · 3747514
1 parent d86ddd9
commit 3747514
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -1,2 +1,3 @@
 colossalai
 torch
+six