chore: fix tests

phil65 · Dec 21, 2024 · b8548c5 · b8548c5
1 parent a011705
commit b8548c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_tokonomics.py b/tests/test_tokonomics.py
@@ -2,13 +2,13 @@
 
 from __future__ import annotations
 
+from pathlib import Path
 import shutil
 
 import diskcache
 import httpx
 import pytest
 import respx
-from upath import UPath
 
 from tokonomics import calculate_token_cost, get_model_costs
 
@@ -25,7 +25,7 @@
 }
 
 # Use a test-specific cache directory
-TEST_CACHE_DIR = UPath("~/.cache/tokonomics/test_pricing").expanduser()
+TEST_CACHE_DIR = Path("~/.cache/tokonomics/test_pricing").expanduser()
 
 
 @pytest.fixture(autouse=True)