Skip to content

Commit 68b2b1b

Browse files
committed
Address last feedback
1 parent bde647e commit 68b2b1b

File tree

1 file changed

+4
-4
lines changed
  • nemoguardrails/llm/cache

1 file changed

+4
-4
lines changed

nemoguardrails/llm/cache/lfu.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def append(self, node: LFUNode) -> None:
5454
"""Add node to the end of the list (before tail)."""
5555
node.prev = self.tail.prev
5656
node.next = self.tail
57-
if self.tail.prev:
57+
if self.tail.prev is not None:
5858
self.tail.prev.next = node
5959
self.tail.prev = node
6060
self.size += 1
@@ -68,9 +68,9 @@ def pop(self, node: Optional[LFUNode] = None) -> Optional[LFUNode]:
6868
node = self.head.next
6969

7070
# Remove node from the list
71-
if node and node.prev:
71+
if node is not None and node.prev is not None:
7272
node.prev.next = node.next
73-
if node and node.next:
73+
if node is not None and node.next is not None:
7474
node.next.prev = node.prev
7575
self.size -= 1
7676

@@ -276,7 +276,7 @@ def get_stats(self) -> dict:
276276

277277
# Calculate hit rate
278278
total_requests = stats["hits"] + stats["misses"]
279-
stats["hit_rate"] = float(
279+
stats["hit_rate"] = (
280280
stats["hits"] / total_requests if total_requests > 0 else 0.0
281281
)
282282

0 commit comments

Comments
 (0)