@@ -28,32 +28,32 @@ def reset_state_fixture() -> None:
2828
2929def test_base_url_option () -> None :
3030 assert replicate .base_url is None
31- assert replicate .client ._client .base_url == URL ("https://api.replicate.com/v1/" )
31+ assert replicate .collections ._client .base_url == URL ("https://api.replicate.com/v1/" )
3232
3333 replicate .base_url = "http://foo.com"
3434
3535 assert replicate .base_url == URL ("http://foo.com" )
36- assert replicate .client ._client .base_url == URL ("http://foo.com" )
36+ assert replicate .collections ._client .base_url == URL ("http://foo.com" )
3737
3838
3939def test_timeout_option () -> None :
4040 assert replicate .timeout == replicate .DEFAULT_TIMEOUT
41- assert replicate .client ._client .timeout == replicate .DEFAULT_TIMEOUT
41+ assert replicate .collections ._client .timeout == replicate .DEFAULT_TIMEOUT
4242
4343 replicate .timeout = 3
4444
4545 assert replicate .timeout == 3
46- assert replicate .client ._client .timeout == 3
46+ assert replicate .collections ._client .timeout == 3
4747
4848
4949def test_max_retries_option () -> None :
5050 assert replicate .max_retries == replicate .DEFAULT_MAX_RETRIES
51- assert replicate .client ._client .max_retries == replicate .DEFAULT_MAX_RETRIES
51+ assert replicate .collections ._client .max_retries == replicate .DEFAULT_MAX_RETRIES
5252
5353 replicate .max_retries = 1
5454
5555 assert replicate .max_retries == 1
56- assert replicate .client ._client .max_retries == 1
56+ assert replicate .collections ._client .max_retries == 1
5757
5858
5959def test_default_headers_option () -> None :
@@ -62,26 +62,26 @@ def test_default_headers_option() -> None:
6262 replicate .default_headers = {"Foo" : "Bar" }
6363
6464 assert replicate .default_headers ["Foo" ] == "Bar"
65- assert replicate .client ._client .default_headers ["Foo" ] == "Bar"
65+ assert replicate .collections ._client .default_headers ["Foo" ] == "Bar"
6666
6767
6868def test_default_query_option () -> None :
6969 assert replicate .default_query is None
70- assert replicate .client ._client ._custom_query == {}
70+ assert replicate .collections ._client ._custom_query == {}
7171
7272 replicate .default_query = {"Foo" : {"nested" : 1 }}
7373
7474 assert replicate .default_query ["Foo" ] == {"nested" : 1 }
75- assert replicate .client ._client ._custom_query ["Foo" ] == {"nested" : 1 }
75+ assert replicate .collections ._client ._custom_query ["Foo" ] == {"nested" : 1 }
7676
7777
7878def test_http_client_option () -> None :
7979 assert replicate .http_client is None
8080
81- original_http_client = replicate .client ._client ._client
81+ original_http_client = replicate .collections ._client ._client
8282 assert original_http_client is not None
8383
8484 new_client = httpx .Client ()
8585 replicate .http_client = new_client
8686
87- assert replicate .client ._client ._client is new_client
87+ assert replicate .collections ._client ._client is new_client
0 commit comments