From e41041de2a3a67a4b7ac0700c0890e9d372f884d Mon Sep 17 00:00:00 2001
From: superjom <superjom@gmail.com>
Date: Tue, 2 Jan 2018 11:48:17 +0800
Subject: [PATCH] fix test

---
 server/visualdl/lib_test.py |  2 +-
 visualdl/logic/pybind.cc    | 17 +++++++----------
 2 files changed, 8 insertions(+), 11 deletions(-)

diff --git a/server/visualdl/lib_test.py b/server/visualdl/lib_test.py
index 88f3cd68..45a5fde9 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 c0b8de80..e78e4ab4 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<T>(std::move(tablet));          \
+    auto tablet = self.tablet(tag);                                        \
+    return vs::components::ScalarReader<T>(std::move(tablet));             \
   })
   py::class_<vs::LogReader>(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<T>(tablet);                                       \
+    auto tablet = self.AddTablet(tag);                                     \
+    return cp::Scalar<T>(tablet);                                          \
   })
 
   py::class_<vs::LogWriter>(m, "LogWriter")
@@ -93,10 +93,7 @@ PYBIND11_PLUGIN(core) {
 
   py::class_<cp::ImageReader::ImageRecord>(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",
-- 
GitLab