Skip to content

Commit d8e0fa3

Browse files
rust bindings: fix expose Selection to python. missed a bit (#20)
Summary: Pull Request resolved: #20 Pull Request resolved: #19 should have been in D74971945 looks like Reviewed By: benjipelletier, mariusae Differential Revision: D75147364
1 parent 4143c07 commit d8e0fa3

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

python/monarch/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
from monarch.common.opaque_ref import OpaqueRef
4646
from monarch.common.pipe import create_pipe, Pipe, remote_generator
4747
from monarch.common.remote import remote
48+
from monarch.common.selection import Selection
4849
from monarch.common.shape import NDSlice, Shape
4950
from monarch.common.stream import get_active_stream, Stream
5051
from monarch.common.tensor import reduce, reduce_, Tensor
@@ -76,6 +77,7 @@
7677
"RemoteException": ("monarch.common.invocation", "RemoteException"),
7778
"Shape": ("monarch.common.shape", "Shape"),
7879
"NDSlice": ("monarch.common.shape", "NDSlice"),
80+
"Selection": ("monarch.common.selection", "Selection"),
7981
"OpaqueRef": ("monarch.common.opaque_ref", "OpaqueRef"),
8082
"create_pipe": ("monarch.common.pipe", "create_pipe"),
8183
"Pipe": ("monarch.common.pipe", "Pipe"),
@@ -143,6 +145,7 @@ def __getattr__(name):
143145
"Future",
144146
"RemoteException",
145147
"Shape",
148+
"Selection",
146149
"NDSlice",
147150
"OpaqueRef",
148151
"create_pipe",

python/monarch/common/selection.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
from monarch._rust_bindings.monarch_hyperactor.selection import Selection
2+
3+
__all__ = ["Selection"]

0 commit comments

Comments
 (0)