fix types

fal-ai · Oct 30, 2024 · f2d3250 · f2d3250
1 parent e7f29dd
commit f2d3250
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/projects/fal/tests/cli/test_deploy.py b/projects/fal/tests/cli/test_deploy.py
@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Optional, Tuple
 from unittest.mock import MagicMock, patch
 
 import pytest
@@ -30,7 +30,7 @@ def mock_parse_pyproject_toml():
 
 
 def mock_args(
-    app_ref: tuple[str],
+    app_ref: Tuple[str, Optional[str]],
     app_name: Optional[str] = None,
     auth: Optional[str] = None,
     strategy: Optional[str] = None,

diff --git a/projects/fal_client/src/fal_client/client.py b/projects/fal_client/src/fal_client/client.py
@@ -10,7 +10,7 @@
 from datetime import datetime, timezone
 from dataclasses import dataclass, field
 from functools import cached_property
-from typing import Any, AsyncIterator, Iterator, TYPE_CHECKING, Optional, Literal
+from typing import Any, AsyncIterator, Dict, Iterator, TYPE_CHECKING, Optional, Literal
 from urllib.parse import urlencode
 
 import httpx
@@ -20,7 +20,7 @@
 if TYPE_CHECKING:
     from PIL import Image
 
-AnyJSON = dict[str, Any]
+AnyJSON = Dict[str, Any]
 Priority = Literal["normal", "low"]
 
 RUN_URL_FORMAT = f"https://{FAL_RUN_HOST}/"