Skip to content

Commit 35e66dc

Browse files
committed
Fix typing of Function create()
1 parent bad0ce4 commit 35e66dc

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

replicate/use.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from dataclasses import dataclass
99
from functools import cached_property
1010
from pathlib import Path
11-
from typing import Any, Dict, Iterator, Optional, Tuple, Union
11+
from typing import Any, Iterator, Optional, Tuple, Union
1212
from urllib.parse import urlparse
1313

1414
import httpx
@@ -333,11 +333,11 @@ def _version(self) -> Version | None:
333333

334334
return version
335335

336-
def __call__(self, **inputs: Dict[str, Any]) -> Any:
336+
def __call__(self, **inputs: Any) -> Any:
337337
run = self.create(**inputs)
338338
return run.output()
339339

340-
def create(self, **inputs: Dict[str, Any]) -> Run:
340+
def create(self, **inputs: Any) -> Run:
341341
"""
342342
Start a prediction with the specified inputs.
343343
"""

0 commit comments

Comments
 (0)