Skip to content

Commit 0901071

Browse files
pzhan9facebook-github-bot
authored andcommitted
Add check for _PythonActorMeshAdapter and _PythonActorMeshRefAdapter (meta-pytorch#777)
Summary: Pull Request resolved: meta-pytorch#777 Add 2 more checks so I can feel warm and fuzzy. Reviewed By: moonli Differential Revision: D79732646 fbshipit-source-id: b48a44ce7aefb2dade85c55b423e180fedf0d352
1 parent a04a812 commit 0901071

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

python/monarch/_src/actor/actor_mesh.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -226,6 +226,10 @@ class _PythonActorMeshAdapter(ActorMeshProtocol):
226226
"""
227227

228228
def __init__(self, inner: PythonActorMesh, proc_mesh: "ProcMesh") -> None:
229+
if _use_standin_mesh():
230+
raise ValueError(
231+
"_PythonActorMeshAdapter should only be used when USE_STANDIN_ACTOR_MESH is not set"
232+
)
229233
self._inner = inner
230234
self._proc_mesh = proc_mesh
231235

@@ -275,6 +279,10 @@ def __init__(
275279
inner: PythonActorMeshRef,
276280
proc_mesh: "Optional[ProcMesh]",
277281
) -> None:
282+
if _use_standin_mesh():
283+
raise ValueError(
284+
"_PythonActorMeshRefAdapter should only be used when USE_STANDIN_ACTOR_MESH is not set"
285+
)
278286
self._inner = inner
279287
self._proc_mesh = proc_mesh
280288

0 commit comments

Comments
 (0)