Skip to content

Commit 2f30126

Browse files
authored
Merge pull request pandas-dev#795 from shashank88/keep_min
Allow passing keep_mins in write and append
2 parents 37a744d + 13c56a4 commit 2f30126

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

arctic/store/version_store.py

+11-3
Original file line numberDiff line numberDiff line change
@@ -606,8 +606,12 @@ def append(self, symbol, data, metadata=None, prune_previous_version=True, upser
606606

607607
if prune_previous_version and previous_version:
608608
# Does not allow prune to remove the base of the new version
609-
self._prune_previous_versions(symbol, keep_version=version.get('base_version_id'),
610-
new_version_shas=version.get(FW_POINTERS_REFS_KEY))
609+
self._prune_previous_versions(
610+
symbol,
611+
keep_version=version.get('base_version_id'),
612+
new_version_shas=version.get(FW_POINTERS_REFS_KEY),
613+
keep_mins=kwargs.get('keep_mins', 120)
614+
)
611615

612616
# Insert the new version into the version DB
613617
version['version'] = next_ver
@@ -658,7 +662,11 @@ def write(self, symbol, data, metadata=None, prune_previous_version=True, **kwar
658662
handler.write(self._arctic_lib, version, symbol, data, previous_version, **kwargs)
659663

660664
if prune_previous_version and previous_version:
661-
self._prune_previous_versions(symbol, new_version_shas=version.get(FW_POINTERS_REFS_KEY))
665+
self._prune_previous_versions(
666+
symbol,
667+
keep_mins=kwargs.get('keep_mins', 120),
668+
new_version_shas=version.get(FW_POINTERS_REFS_KEY)
669+
)
662670

663671
# Insert the new version into the version DB
664672
self._insert_version(version)

0 commit comments

Comments
 (0)