Pass tests

expectedparrot · Jan 18, 2025 · 35c0f08 · 35c0f08
1 parent 130f8a6
commit 35c0f08
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/edsl/language_models/key_management/KeyLookupBuilder.py b/edsl/language_models/key_management/KeyLookupBuilder.py
@@ -36,7 +36,7 @@ class KeyLookupBuilder:
 
     >>> builder = KeyLookupBuilder(fetch_order=("config", "env"))
     >>> builder.DEFAULT_RPM
-    10
+    100
     >>> builder.DEFAULT_TPM
     2000000
     >>> builder.fetch_order
@@ -58,8 +58,8 @@ class KeyLookupBuilder:
     # DEFAULT_TPM = 2000000
     from edsl.config import CONFIG
 
-    DEFAULT_RPM = CONFIG.get("EDSL_SERVICE_RPM_BASELINE")
-    DEFAULT_TPM = CONFIG.get("EDSL_SERVICE_TPM_BASELINE")
+    DEFAULT_RPM = int(CONFIG.get("EDSL_SERVICE_RPM_BASELINE"))
+    DEFAULT_TPM = int(CONFIG.get("EDSL_SERVICE_TPM_BASELINE"))
 
     def __init__(self, fetch_order: Optional[tuple[str]] = None):
         if fetch_order is None:

diff --git a/tests/language_models/test_KeyLookupBuilder.py b/tests/language_models/test_KeyLookupBuilder.py
@@ -137,8 +137,8 @@ def test_default_limits():
 
     result = builder.get_language_model_input("test")
 
-    assert result.rpm == builder.DEFAULT_RPM
-    assert result.tpm == builder.DEFAULT_TPM
+    assert float(result.rpm) == float(builder.DEFAULT_RPM)
+    assert float(result.tpm) == float(builder.DEFAULT_TPM)
 
 
 def test_build_method():