Skip to content

Commit 14f6cfc

Browse files
chore: use lazy imports for module level client
1 parent 7dc2438 commit 14f6cfc

File tree

1 file changed

+38
-25
lines changed

1 file changed

+38
-25
lines changed

src/replicate/_module_client.py

Lines changed: 38 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -1,64 +1,77 @@
11
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
22

3+
from __future__ import annotations
4+
5+
from typing import TYPE_CHECKING
36
from typing_extensions import override
47

5-
from . import resources, _load_client
8+
if TYPE_CHECKING:
9+
from .resources.account import AccountResource
10+
from .resources.hardware import HardwareResource
11+
from .resources.trainings import TrainingsResource
12+
from .resources.collections import CollectionsResource
13+
from .resources.predictions import PredictionsResource
14+
from .resources.models.models import ModelsResource
15+
from .resources.webhooks.webhooks import WebhooksResource
16+
from .resources.deployments.deployments import DeploymentsResource
17+
18+
from . import _load_client
619
from ._utils import LazyProxy
720

821

9-
class ModelsResourceProxy(LazyProxy[resources.ModelsResource]):
22+
class ModelsResourceProxy(LazyProxy["ModelsResource"]):
1023
@override
11-
def __load__(self) -> resources.ModelsResource:
24+
def __load__(self) -> ModelsResource:
1225
return _load_client().models
1326

1427

15-
class AccountResourceProxy(LazyProxy[resources.AccountResource]):
28+
class AccountResourceProxy(LazyProxy["AccountResource"]):
1629
@override
17-
def __load__(self) -> resources.AccountResource:
30+
def __load__(self) -> AccountResource:
1831
return _load_client().account
1932

2033

21-
class HardwareResourceProxy(LazyProxy[resources.HardwareResource]):
34+
class HardwareResourceProxy(LazyProxy["HardwareResource"]):
2235
@override
23-
def __load__(self) -> resources.HardwareResource:
36+
def __load__(self) -> HardwareResource:
2437
return _load_client().hardware
2538

2639

27-
class WebhooksResourceProxy(LazyProxy[resources.WebhooksResource]):
40+
class WebhooksResourceProxy(LazyProxy["WebhooksResource"]):
2841
@override
29-
def __load__(self) -> resources.WebhooksResource:
42+
def __load__(self) -> WebhooksResource:
3043
return _load_client().webhooks
3144

3245

33-
class TrainingsResourceProxy(LazyProxy[resources.TrainingsResource]):
46+
class TrainingsResourceProxy(LazyProxy["TrainingsResource"]):
3447
@override
35-
def __load__(self) -> resources.TrainingsResource:
48+
def __load__(self) -> TrainingsResource:
3649
return _load_client().trainings
3750

3851

39-
class CollectionsResourceProxy(LazyProxy[resources.CollectionsResource]):
52+
class CollectionsResourceProxy(LazyProxy["CollectionsResource"]):
4053
@override
41-
def __load__(self) -> resources.CollectionsResource:
54+
def __load__(self) -> CollectionsResource:
4255
return _load_client().collections
4356

4457

45-
class DeploymentsResourceProxy(LazyProxy[resources.DeploymentsResource]):
58+
class DeploymentsResourceProxy(LazyProxy["DeploymentsResource"]):
4659
@override
47-
def __load__(self) -> resources.DeploymentsResource:
60+
def __load__(self) -> DeploymentsResource:
4861
return _load_client().deployments
4962

5063

51-
class PredictionsResourceProxy(LazyProxy[resources.PredictionsResource]):
64+
class PredictionsResourceProxy(LazyProxy["PredictionsResource"]):
5265
@override
53-
def __load__(self) -> resources.PredictionsResource:
66+
def __load__(self) -> PredictionsResource:
5467
return _load_client().predictions
5568

5669

57-
models: resources.ModelsResource = ModelsResourceProxy().__as_proxied__()
58-
account: resources.AccountResource = AccountResourceProxy().__as_proxied__()
59-
hardware: resources.HardwareResource = HardwareResourceProxy().__as_proxied__()
60-
webhooks: resources.WebhooksResource = WebhooksResourceProxy().__as_proxied__()
61-
trainings: resources.TrainingsResource = TrainingsResourceProxy().__as_proxied__()
62-
collections: resources.CollectionsResource = CollectionsResourceProxy().__as_proxied__()
63-
deployments: resources.DeploymentsResource = DeploymentsResourceProxy().__as_proxied__()
64-
predictions: resources.PredictionsResource = PredictionsResourceProxy().__as_proxied__()
70+
models: ModelsResource = ModelsResourceProxy().__as_proxied__()
71+
account: AccountResource = AccountResourceProxy().__as_proxied__()
72+
hardware: HardwareResource = HardwareResourceProxy().__as_proxied__()
73+
webhooks: WebhooksResource = WebhooksResourceProxy().__as_proxied__()
74+
trainings: TrainingsResource = TrainingsResourceProxy().__as_proxied__()
75+
collections: CollectionsResource = CollectionsResourceProxy().__as_proxied__()
76+
deployments: DeploymentsResource = DeploymentsResourceProxy().__as_proxied__()
77+
predictions: PredictionsResource = PredictionsResourceProxy().__as_proxied__()

0 commit comments

Comments
 (0)