Skip to content

Commit 571f272

Browse files
committed
Fix the nil memory eviction when using the init_similar_cache method
Signed-off-by: SimFG <bang.fu@zilliz.com>
1 parent 75ab7ec commit 571f272

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

gptcache/manager/factory.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,12 @@ def manager_factory(manager="map",
118118
maxmemory_samples=eviction_params.get("maxmemory_samples", scalar_params.get("maxmemory_samples")),
119119
)
120120

121+
if eviction_manager == "memory":
122+
return get_data_manager(s, v, o, None,
123+
eviction_params.get("max_size", 1000),
124+
eviction_params.get("clean_size", None),
125+
eviction_params.get("eviction", "LRU"),)
126+
121127
e = EvictionBase(
122128
name=eviction_manager,
123129
**eviction_params
@@ -194,7 +200,7 @@ def get_data_manager(
194200
vector_base = VectorBase(name=vector_base)
195201
if isinstance(object_base, str):
196202
object_base = ObjectBase(name=object_base)
197-
if isinstance(eviction_base, str):
203+
if isinstance(eviction_base, str) and eviction_base != "memory":
198204
eviction_base = EvictionBase(name=eviction_base)
199205
assert cache_base and vector_base
200206
return SSDataManager(cache_base, vector_base, object_base, eviction_base, max_size, clean_size, eviction)

0 commit comments

Comments
 (0)