add/set default partition size

PyQuantSharp · Sep 27, 2020 · 2171936 · 2171936
1 parent e2954e8
commit 2171936
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pystore/utils.py b/pystore/utils.py
@@ -151,3 +151,13 @@ def delete_store(store):
 def delete_stores():
     shutil.rmtree(get_path())
     return True
+
+def set_partition_size(size=None):
+    if size is None:
+        size = config.DEFAULT_PARTITION_SIZE * 1
+    config.PARTITION_SIZE = size
+    return config.PARTITION_SIZE
+
+
+def get_partition_size():
+    return config.PARTITION_SIZE