fix test

pintaoz · pintaoz · commit 49cd52ebb084 · 2025-02-20T15:03:39.000-08:00
diff --git a/src/sagemaker/amazon/common.py b/src/sagemaker/amazon/common.py
@@ -21,4 +21,5 @@
     read_records,
     write_numpy_to_dense_tensor,
     write_spmatrix_to_sparse_tensor,
+    _write_recordio,
 )

Original file line number	Diff line number	Diff line change
`@@ -21,4 +21,5 @@`
`21`	`21`	`read_records,`
`22`	`22`	`write_numpy_to_dense_tensor,`
`23`	`23`	`write_spmatrix_to_sparse_tensor,`
	`24`	`+ _write_recordio,`
`24`	`25`	`)`