Skip to content

Commit 97b771a

Browse files
fix(package): support direct resource imports
1 parent 07654af commit 97b771a

File tree

2 files changed

+28
-0
lines changed

2 files changed

+28
-0
lines changed

src/replicate/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
from __future__ import annotations
44

5+
import typing as _t
56
from typing_extensions import override
67

78
from . import types
@@ -89,6 +90,9 @@
8990
"ModelVersionIdentifier",
9091
]
9192

93+
if not _t.TYPE_CHECKING:
94+
from ._utils._resources_proxy import resources as resources
95+
9296
_setup_logging()
9397

9498
# Update the __module__ attribute for exported symbols so that
Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
from __future__ import annotations
2+
3+
from typing import Any
4+
from typing_extensions import override
5+
6+
from ._proxy import LazyProxy
7+
8+
9+
class ResourcesProxy(LazyProxy[Any]):
10+
"""A proxy for the `replicate.resources` module.
11+
12+
This is used so that we can lazily import `replicate.resources` only when
13+
needed *and* so that users can just import `replicate` and reference `replicate.resources`
14+
"""
15+
16+
@override
17+
def __load__(self) -> Any:
18+
import importlib
19+
20+
mod = importlib.import_module("replicate.resources")
21+
return mod
22+
23+
24+
resources = ResourcesProxy().__as_proxied__()

0 commit comments

Comments
 (0)