Remove logger from Database (#3270)

ray6080 · web-flow · commit b3540b729ae5 · 2024-05-09T13:06:03.000-04:00
diff --git a/src_cpp/include/py_database.h b/src_cpp/include/py_database.h
@@ -11,10 +11,6 @@ class PyDatabase {
     friend class PyConnection;
 
 public:
-    inline void setLoggingLevel(std::string logging_level) {
-        database->setLoggingLevel(std::move(logging_level));
-    }
-
     static void initialize(py::handle& m);
 
     static py::str getVersion();
diff --git a/src_cpp/py_database.cpp b/src_cpp/py_database.cpp
@@ -15,7 +15,6 @@ void PyDatabase::initialize(py::handle& m) {
             py::arg("database_path"), py::arg("buffer_pool_size") = 0,
             py::arg("max_num_threads") = 0, py::arg("compression") = true,
             py::arg("read_only") = false, py::arg("max_db_size") = (uint64_t)1 << 43)
-        .def("set_logging_level", &PyDatabase::setLoggingLevel, py::arg("logging_level"))
         .def("scan_node_table_as_int64", &PyDatabase::scanNodeTable<std::int64_t>,
             py::arg("table_name"), py::arg("prop_name"), py::arg("indices"), py::arg("np_array"),
             py::arg("num_threads"))
diff --git a/src_py/database.py b/src_py/database.py
@@ -26,15 +26,15 @@ class Database:
     """Kùzu database instance."""
 
     def __init__(
-        self,
-        database_path: str | Path,
-        *,
-        buffer_pool_size: int = 0,
-        max_num_threads: int = 0,
-        compression: bool = True,
-        lazy_init: bool = False,
-        read_only: bool = False,
-        max_db_size: int = (1 << 43),
+            self,
+            database_path: str | Path,
+            *,
+            buffer_pool_size: int = 0,
+            max_num_threads: int = 0,
+            compression: bool = True,
+            lazy_init: bool = False,
+            read_only: bool = False,
+            max_db_size: int = (1 << 43),
     ):
         """
         Parameters
@@ -92,10 +92,10 @@ def __enter__(self) -> Self:
         return self
 
     def __exit__(
-        self,
-        exc_type: type[BaseException] | None,
-        exc_value: BaseException | None,
-        exc_traceback: TracebackType | None,
+            self,
+            exc_type: type[BaseException] | None,
+            exc_value: BaseException | None,
+            exc_traceback: TracebackType | None,
     ) -> None:
         self.close()
 
@@ -146,21 +146,8 @@ def init_database(self) -> None:
                 self.max_db_size,
             )
 
-    def set_logging_level(self, level: str) -> None:
-        """
-        Set the logging level.
-
-        Parameters
-        ----------
-        level : str
-            Logging level. One of "debug", "info", "err".
-
-        """
-        self.check_for_database_close()
-        self._database.set_logging_level(level)
-
     def get_torch_geometric_remote_backend(
-        self, num_threads: int | None = None
+            self, num_threads: int | None = None
     ) -> tuple[KuzuFeatureStore, KuzuGraphStore]:
         """
         Use the database as the remote backend for torch_geometric.
@@ -213,13 +200,13 @@ def get_torch_geometric_remote_backend(
         )
 
     def _scan_node_table(
-        self,
-        table_name: str,
-        prop_name: str,
-        prop_type: str,
-        dim: int,
-        indices: IndexType,
-        num_threads: int,
+            self,
+            table_name: str,
+            prop_name: str,
+            prop_type: str,
+            dim: int,
+            indices: IndexType,
+            num_threads: int,
     ) -> NDArray[Any]:
         self.check_for_database_close()
         import numpy as np
