Merge pull request #309 from koterpillar/cache-root-driver

chore(test): cache root driver
koterpillar · Jan 9, 2025 · 2f1ab1f · 2f1ab1f
2 parents ee3acf2 + ca8ed0b
commit 2f1ab1f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mybox/driver.py b/mybox/driver.py
@@ -3,6 +3,7 @@
 from collections.abc import AsyncIterator, Callable, Iterable
 from contextlib import AbstractAsyncContextManager, asynccontextmanager, nullcontext
 from dataclasses import dataclass
+from functools import cache
 from os import environ
 from pathlib import Path
 from typing import Any, Literal, Optional, cast
@@ -70,6 +71,7 @@ def __init__(self, *, root: bool = False) -> None:
     def deconstruct(self) -> dict:
         return {"root": self.root}
 
+    @cache
     def with_root(self, /, root: bool) -> "Driver":
         if self.root == root:
             return self