diff --git a/server/visualdl/lib_test.py b/server/visualdl/lib_test.py index 88f3cd68f30c2de9eec8e024c1f58594d66ee8f1..45a5fde945df2dec1466a33aafe6a74e04e72bbb 100644 --- a/server/visualdl/lib_test.py +++ b/server/visualdl/lib_test.py @@ -8,7 +8,7 @@ from storage_mock import add_scalar, add_image class LibTest(unittest.TestCase): def setUp(self): 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, "test", "layer/scalar0/min", 1000, 10) diff --git a/visualdl/logic/pybind.cc b/visualdl/logic/pybind.cc index c0b8de8010165fa9baa7de1812cf8b34e8135142..e78e4ab45eccfcd241b6e3cb2c47c8dd75c10fd2 100644 --- a/visualdl/logic/pybind.cc +++ b/visualdl/logic/pybind.cc @@ -11,10 +11,10 @@ namespace cp = visualdl::components; PYBIND11_PLUGIN(core) { py::module m("core", "C++ core of VisualDL"); -#define READER_ADD_SCALAR(T) \ +#define READER_ADD_SCALAR(T) \ .def("get_scalar_" #T, [](vs::LogReader& self, const std::string& tag) { \ - auto tablet = self.tablet(tag); \ - return vs::components::ScalarReader(std::move(tablet)); \ + auto tablet = self.tablet(tag); \ + return vs::components::ScalarReader(std::move(tablet)); \ }) py::class_(m, "LogReader") .def("__init__", @@ -35,10 +35,10 @@ PYBIND11_PLUGIN(core) { }); #undef READER_ADD_SCALAR -#define WRITER_ADD_SCALAR(T) \ +#define WRITER_ADD_SCALAR(T) \ .def("new_scalar_" #T, [](vs::LogWriter& self, const std::string& tag) { \ - auto tablet = self.AddTablet(tag); \ - return cp::Scalar(tablet); \ + auto tablet = self.AddTablet(tag); \ + return cp::Scalar(tablet); \ }) py::class_(m, "LogWriter") @@ -93,10 +93,7 @@ PYBIND11_PLUGIN(core) { py::class_(m, "ImageRecord") // TODO(ChunweiYan) make these copyless. - .def("data", - [](cp::ImageReader::ImageRecord& self) { - return self.data; - }) + .def("data", [](cp::ImageReader::ImageRecord& self) { return self.data; }) .def("shape", [](cp::ImageReader::ImageRecord& self) { return self.shape; }) .def("step_id",