From 019f2b1222975a0571361bda60af62e981aec29b Mon Sep 17 00:00:00 2001
From: Lucas Jeub <lucas.jeub@pometry.com>
Date: Fri, 13 Dec 2024 10:15:15 +0100
Subject: [PATCH] tidy

---
 python/python/raphtory/__init__.pyi            | 17 +++++++++--------
 python/python/raphtory/algorithms/__init__.pyi |  2 +-
 python/tests/test_index.py                     |  1 +
 3 files changed, 11 insertions(+), 9 deletions(-)

diff --git a/python/python/raphtory/__init__.pyi b/python/python/raphtory/__init__.pyi
index bbe2bf8fc..9882a820f 100644
--- a/python/python/raphtory/__init__.pyi
+++ b/python/python/raphtory/__init__.pyi
@@ -136,7 +136,7 @@ class ConstProperties(object):
     """A view of constant properties of an entity"""
 
     def __contains__(self, key):
-        """Return key in self."""
+        """Return bool(key in self)."""
 
     def __eq__(self, value):
         """Return self==value."""
@@ -220,11 +220,12 @@ class DiskGraphStorage(object):
     def load_from_parquets(
         graph_dir,
         layer_parquet_cols,
-        node_properties,
-        chunk_size,
-        t_props_chunk_size,
-        num_threads,
-        node_type_col,
+        node_properties=None,
+        chunk_size=10000000,
+        t_props_chunk_size=10000000,
+        num_threads=4,
+        node_type_col=None,
+        node_id_col=None,
     ): ...
     def load_node_const_properties(self, location, col_names=None, chunk_size=None): ...
     def merge_by_sorted_gids(self, other, graph_dir):
@@ -4498,7 +4499,7 @@ class Properties(object):
     """A view of the properties of an entity"""
 
     def __contains__(self, key):
-        """Return key in self."""
+        """Return bool(key in self)."""
 
     def __eq__(self, value):
         """Return self==value."""
@@ -4685,7 +4686,7 @@ class TemporalProperties(object):
     """A view of the temporal properties of an entity"""
 
     def __contains__(self, key):
-        """Return key in self."""
+        """Return bool(key in self)."""
 
     def __eq__(self, value):
         """Return self==value."""
diff --git a/python/python/raphtory/algorithms/__init__.pyi b/python/python/raphtory/algorithms/__init__.pyi
index 278c38cee..84a1c21c7 100644
--- a/python/python/raphtory/algorithms/__init__.pyi
+++ b/python/python/raphtory/algorithms/__init__.pyi
@@ -24,7 +24,7 @@ class Matching(object):
         """True if self else False"""
 
     def __contains__(self, key):
-        """Return key in self."""
+        """Return bool(key in self)."""
 
     def __iter__(self):
         """Implement iter(self)."""
diff --git a/python/tests/test_index.py b/python/tests/test_index.py
index 9843f42fe..e6bf534a2 100644
--- a/python/tests/test_index.py
+++ b/python/tests/test_index.py
@@ -3,6 +3,7 @@
 # Range queries are not supported for properties in tantivy (as of 0.22) because they are implemented as json.
 # See https://github.com/quickwit-oss/tantivy/issues/1709
 
+
 def test_search_in_python():
     g = Graph()
     g.add_node(