Skip to content

Commit 1c5f391

Browse files
committed
Cleanup after release
1 parent 8fe81c5 commit 1c5f391

File tree

2 files changed

+4
-14
lines changed

2 files changed

+4
-14
lines changed

superduper/backends/simple/cluster.py

Lines changed: 3 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,8 @@
55

66
from superduper import logging
77
from superduper.backends.base.cluster import Cluster
8-
from superduper.backends.local.cache import LocalCache
98
from superduper.backends.local.cdc import LocalCDCBackend
10-
from superduper.backends.simple.compute import SimpleComputeBackend, SimpleComputeClient
9+
from superduper.backends.simple.compute import SimpleComputeClient
1110
from superduper.backends.local.crontab import LocalCrontabBackend
1211
from superduper.backends.simple.scheduler import SimpleScheduler
1312
from superduper.backends.local.vector_search import LocalVectorSearchBackend
@@ -29,15 +28,8 @@ class SimpleCluster(Cluster):
2928
def build(cls, CFG, **kwargs):
3029
"""Build the local cluster."""
3130
searcher_impl = load_plugin(CFG.vector_search_engine).VectorSearcher
32-
cache = None
33-
if CFG.cache and CFG.cache.startswith('redis'):
34-
cache = load_plugin('redis').Cache(uri=CFG.cache)
35-
elif CFG.cache:
36-
assert CFG.cache == 'in-process'
37-
cache = LocalCache()
3831

3932
return SimpleCluster(
40-
cache=cache,
4133
scheduler=SimpleScheduler(),
4234
compute=SimpleComputeClient(),
4335
vector_search=LocalVectorSearchBackend(searcher_impl=searcher_impl),
@@ -52,9 +44,7 @@ def drop(self, force: bool = False):
5244
"""
5345
if not force:
5446
if not click.confirm(
55-
"Are you sure you want to drop the cache? ",
47+
"Are you sure you want to drop the cluster? ",
5648
default=False,
5749
):
58-
logging.warn("Aborting...")
59-
if self.cache is not None:
60-
return self.cache.drop()
50+
logging.warn("Aborting...")

superduper/backends/simple/scheduler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ class QueueWrapper:
2222
:param scheduler: LocalScheduler instance.
2323
"""
2424

25-
def __init__(self, identifier: str, scheduler: 'LocalScheduler'):
25+
def __init__(self, identifier: str, scheduler: 'SimpleScheduler'):
2626
self.scheduler = scheduler
2727
self.identifier = identifier
2828

0 commit comments

Comments
 (0)