提交 e41041de 编写于 作者: S superjom

fix test

上级 31a2a715
...@@ -8,7 +8,7 @@ from storage_mock import add_scalar, add_image ...@@ -8,7 +8,7 @@ from storage_mock import add_scalar, add_image
class LibTest(unittest.TestCase): class LibTest(unittest.TestCase):
def setUp(self): def setUp(self):
dir = "./tmp/mock" dir = "./tmp/mock"
writer = storage.StorageWriter(dir, sync_cycle=20) writer = storage.LogWriter(dir, sync_cycle=20)
add_scalar(writer, "train", "layer/scalar0/min", 1000, 1) add_scalar(writer, "train", "layer/scalar0/min", 1000, 1)
add_scalar(writer, "test", "layer/scalar0/min", 1000, 10) add_scalar(writer, "test", "layer/scalar0/min", 1000, 10)
......
...@@ -93,10 +93,7 @@ PYBIND11_PLUGIN(core) { ...@@ -93,10 +93,7 @@ PYBIND11_PLUGIN(core) {
py::class_<cp::ImageReader::ImageRecord>(m, "ImageRecord") py::class_<cp::ImageReader::ImageRecord>(m, "ImageRecord")
// TODO(ChunweiYan) make these copyless. // TODO(ChunweiYan) make these copyless.
.def("data", .def("data", [](cp::ImageReader::ImageRecord& self) { return self.data; })
[](cp::ImageReader::ImageRecord& self) {
return self.data;
})
.def("shape", .def("shape",
[](cp::ImageReader::ImageRecord& self) { return self.shape; }) [](cp::ImageReader::ImageRecord& self) { return self.shape; })
.def("step_id", .def("step_id",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册