Skip to content

Commit ff8964b

Browse files
Fixed linting errors
1 parent a55e288 commit ff8964b

6 files changed

Lines changed: 20 additions & 15 deletions

File tree

vectorq/config.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,9 @@
1515
HNSWLibVectorDB,
1616
)
1717
from vectorq.vectorq_core.cache.eviction_policy.eviction_policy import EvictionPolicy
18-
from vectorq.vectorq_core.cache.eviction_policy.strategies.no_eviction import NoEvictionPolicy
18+
from vectorq.vectorq_core.cache.eviction_policy.strategies.no_eviction import (
19+
NoEvictionPolicy,
20+
)
1921

2022

2123
class VectorQConfig:

vectorq/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def __init__(
2020
self.vectorq_config = config
2121
self.vectorq_policy = policy
2222
self.vectorq_policy.setup(config)
23-
23+
2424
def infer(
2525
self,
2626
prompt: str,

vectorq/vectorq_core/cache/eviction_policy/__init__.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
1-
from vectorq.vectorq_core.cache.eviction_policy.strategies.lru import LRUEvictionPolicy
2-
from vectorq.vectorq_core.cache.eviction_policy.strategies.no_eviction import NoEvictionPolicy
31
from vectorq.vectorq_core.cache.eviction_policy.eviction_policy import EvictionPolicy
2+
from vectorq.vectorq_core.cache.eviction_policy.strategies.lru import LRUEvictionPolicy
3+
from vectorq.vectorq_core.cache.eviction_policy.strategies.no_eviction import (
4+
NoEvictionPolicy,
5+
)
46

57
__all__ = [
68
"LRUEvictionPolicy",

vectorq/vectorq_core/cache/eviction_policy/strategies/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
from vectorq.vectorq_core.cache.eviction_policy.strategies.lru import LRUEvictionPolicy
2-
from vectorq.vectorq_core.cache.eviction_policy.strategies.no_eviction import NoEvictionPolicy
2+
from vectorq.vectorq_core.cache.eviction_policy.strategies.no_eviction import (
3+
NoEvictionPolicy,
4+
)
35

46
__all__ = [
57
"LRUEvictionPolicy",

vectorq/vectorq_policy/strategies/dynamic_global_threshold.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
from enum import Enum
33
from typing import Dict, List, Optional, Tuple
44

5-
import cvxpy as cp
65
import numpy as np
76
import statsmodels.api as sm
87
from scipy.special import expit
@@ -120,10 +119,10 @@ def __init__(self, delta: float):
120119
self.global_var_t: float = None
121120

122121
self.variance_map: Dict[int, List[float]] = {
123-
6: 0.035445,
124-
7: 0.028285,
125-
8: 0.026436,
126-
9: 0.021349,
122+
6: 0.035445,
123+
7: 0.028285,
124+
8: 0.026436,
125+
9: 0.021349,
127126
10: 0.019371,
128127
11: 0.012615,
129128
12: 0.011433,
@@ -244,7 +243,7 @@ def _estimate_parameters(
244243
print(f"len does not match: {len(similarities)} != {len(labels)}")
245244
self.logistic_regression.fit(similarities, labels)
246245
intercept, gamma = self.logistic_regression.coef_[0]
247-
246+
248247
gamma = max(gamma, 1e-6)
249248
t_hat = -intercept / gamma
250249
t_hat = float(np.clip(t_hat, 0.0, 1.0))

vectorq/vectorq_policy/strategies/dynamic_local_threshold.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -111,10 +111,10 @@ def __init__(self, delta: float):
111111
)
112112

113113
self.variance_map: Dict[int, List[float]] = {
114-
6: 0.035445,
115-
7: 0.028285,
116-
8: 0.026436,
117-
9: 0.021349,
114+
6: 0.035445,
115+
7: 0.028285,
116+
8: 0.026436,
117+
9: 0.021349,
118118
10: 0.019371,
119119
11: 0.012615,
120120
12: 0.011433,

0 commit comments

Comments
 (0)