fix: upgrade Pydantic req; more accurate import source in test

fal-ai · Feb 11, 2024 · 8ac53ff · 8ac53ff
1 parent ec3d302
commit 8ac53ff
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/projects/fal/src/fal/toolkit/file/providers/__init__.py b/projects/fal/src/fal/toolkit/file/providers/__init__.py
@@ -0,0 +1,10 @@
+from .fal import FalFileRepository, InMemoryRepository
+from .gcp import GoogleStorageRepository
+from .r2 import R2Repository
+
+__all__ = [
+    "FalFileRepository",
+    "InMemoryRepository",
+    "GoogleStorageRepository",
+    "R2Repository",
+]
diff --git a/projects/fal/tests/stability_test.py b/projects/fal/tests/stability_test.py
@@ -419,7 +419,7 @@ class MathQuery(BaseModel):
     class MathResult(BaseModel):
         result: int = Field(description="The result of the operation")
 
-    @isolated_client("virtualenv", requirements=["pydantic<2"])
+    @isolated_client("virtualenv", requirements=["pydantic>=2"])
     def add(query: MathQuery) -> MathResult:
         return MathResult(result=query.x + query.y)
 

diff --git a/projects/fal/tests/toolkit/file_test.py b/projects/fal/tests/toolkit/file_test.py
@@ -4,7 +4,8 @@
 from base64 import b64encode
 
 import pytest
-from fal.toolkit.file.file import File, GoogleStorageRepository
+from fal.toolkit.file.file import File
+from fal.toolkit.file.providers import GoogleStorageRepository
 
 
 def test_binary_content_matches():