File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 22
33import os
44import sys
5+ from typing import Any
56from unittest .mock import MagicMock , patch
67
78
@@ -16,7 +17,7 @@ def test_use_does_not_create_client_immediately():
1617 import replicate
1718
1819 # This should work now - no client is created yet
19- model = replicate .use ("test/model" )
20+ model : Any = replicate .use ("test/model" ) # type: ignore[misc]
2021
2122 # Verify we got a Function object back
2223 from replicate .lib ._predictions_use import Function
@@ -26,7 +27,7 @@ def test_use_does_not_create_client_immediately():
2627
2728 # Verify the client property is a property that will create client on demand
2829 # We can't call it without a token, but we can check it's the right type
29- assert hasattr (model , "_client_or_factory" )
30+ assert hasattr (model , "_client_or_factory" ) # type: ignore[misc]
3031 print ("✓ Client factory is stored for lazy creation" )
3132
3233 except Exception as e :
@@ -50,7 +51,7 @@ def test_client_created_when_model_called():
5051 import replicate
5152
5253 # Create model function - should work without errors
53- model = replicate .use ("test/model" )
54+ model : Any = replicate .use ("test/model" ) # type: ignore[misc]
5455 print ("✓ Model function created successfully" )
5556
5657 # Verify the model has the lazy client setup
You can’t perform that action at this time.
0 commit comments