Skip to content

Commit 0f95a92

Browse files
committed
lint
1 parent 6c1c60f commit 0f95a92

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

src/replicate/_module_client.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -89,12 +89,13 @@ def _run(*args, **kwargs):
8989

9090
def _use(ref, *, hint=None, streaming=False, use_async=False, **kwargs):
9191
from .lib._predictions_use import use
92-
92+
9393
if use_async:
9494
# For async, we need to use AsyncReplicate instead
9595
from ._client import AsyncReplicate
96+
9697
return use(lambda: AsyncReplicate(), ref, hint=hint, streaming=streaming, use_async=True, **kwargs)
97-
98+
9899
return use(_load_client, ref, hint=hint, streaming=streaming, use_async=False, **kwargs)
99100

100101
run = _run

tests/test_simple_lazy.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def test_use_does_not_create_client_immediately():
2626

2727
# Verify the client property is a property that will create client on demand
2828
# We can't call it without a token, but we can check it's the right type
29-
assert hasattr(model, '_client_or_factory')
29+
assert hasattr(model, "_client_or_factory")
3030
print("✓ Client factory is stored for lazy creation")
3131

3232
except Exception as e:
@@ -54,7 +54,7 @@ def test_client_created_when_model_called():
5454
print("✓ Model function created successfully")
5555

5656
# Verify the model has the lazy client setup
57-
assert hasattr(model, '_client_or_factory')
57+
assert hasattr(model, "_client_or_factory")
5858
assert callable(model._client_or_factory)
5959
print("✓ Lazy client factory is properly configured")
6060

0 commit comments

Comments
 (0)