Back out "fix broken tests"

Pull Request resolved: #751

Checks to see if the fix below this in the stack actually works.
ghstack-source-id: 300893831

Differential Revision: [D79586849](https://our.internmc.facebook.com/intern/diff/D79586849/)

Author: zdevito

python/tests/test_python_actors.py

@@ -86,8 +86,6 @@ async def test_choose():
     assert_type(result, int)
     assert result2 == result3
 
-    await proc.stop()
-
 
 @pytest.mark.timeout(60)
 async def test_stream():
@@ -97,8 +95,6 @@ async def test_stream():
 
     assert 8 == sum([await x for x in v.value.stream()])
 
-    await proc.stop()
-
 
 class To(Actor):
     @endpoint
@@ -121,8 +117,6 @@ async def test_mesh_passed_to_mesh():
     assert len(all) == 4
     assert all[0] != all[1]
 
-    await proc.stop()
-
 
 @pytest.mark.timeout(60)
 async def test_mesh_passed_to_mesh_on_different_proc_mesh():
@@ -134,9 +128,6 @@ async def test_mesh_passed_to_mesh_on_different_proc_mesh():
     assert len(all) == 4
     assert all[0] != all[1]
 
-    await proc.stop()
-    await proc2.stop()
-
 
 @pytest.mark.timeout(60)
 async def test_actor_slicing():
@@ -153,9 +144,6 @@ async def test_actor_slicing():
 
     assert result[0] == result[1]
 
-    await proc.stop()
-    await proc2.stop()
-
 
 @pytest.mark.timeout(60)
 async def test_aggregate():
@@ -166,8 +154,6 @@ async def test_aggregate():
     r = await acc.accumulate()
     assert r == 4
 
-    await proc.stop()
-
 
 class RunIt(Actor):
     @endpoint
@@ -185,8 +171,6 @@ async def test_rank_size():
     assert 1 == await acc.accumulate(lambda: current_rank()["gpus"])
     assert 4 == await acc.accumulate(lambda: current_size()["gpus"])
 
-    await proc.stop()
-
 
 class SyncActor(Actor):
     @endpoint
@@ -202,51 +186,41 @@ async def test_sync_actor():
     r = await a.sync_endpoint.choose(c)
     assert r == 5
 
-    await proc.stop()
-
 
 @pytest.mark.timeout(60)
-async def test_sync_actor_sync_client() -> None:
+def test_sync_actor_sync_client() -> None:
     proc = local_proc_mesh(gpus=2)
     a = proc.spawn("actor", SyncActor).get()
     c = proc.spawn("counter", Counter, 5).get()
     r = a.sync_endpoint.choose(c).get()
     assert r == 5
 
-    await proc.stop()
-
 
 @pytest.mark.timeout(60)
-async def test_proc_mesh_size() -> None:
+def test_proc_mesh_size() -> None:
     proc = local_proc_mesh(gpus=2)
     assert 2 == proc.size("gpus")
-    # proc.initialized.get()
-    # await proc.stop()
 
 
 @pytest.mark.timeout(60)
-async def test_rank_size_sync() -> None:
+def test_rank_size_sync() -> None:
     proc = local_proc_mesh(gpus=2)
     r = proc.spawn("runit", RunIt).get()
 
     acc = Accumulator(r.run, 0, operator.add)
     assert 1 == acc.accumulate(lambda: current_rank()["gpus"]).get()
     assert 4 == acc.accumulate(lambda: current_size()["gpus"]).get()
 
-    await proc.stop()
-
 
 @pytest.mark.timeout(60)
-async def test_accumulate_sync() -> None:
+def test_accumulate_sync() -> None:
     proc = local_proc_mesh(gpus=2)
     counter = proc.spawn("counter", Counter, 1).get()
     counter.incr.broadcast()
     acc = Accumulator(counter.value, 0, operator.add)
     r = acc.accumulate().get()
     assert r == 4
 
-    await proc.stop()
-
 
 class CastToCounter(Actor):
     @endpoint
@@ -255,7 +229,7 @@ def doit(self, c: Counter):
 
 
 @pytest.mark.timeout(60)
-async def test_value_mesh() -> None:
+def test_value_mesh() -> None:
     proc = local_proc_mesh(gpus=2)
     counter = proc.spawn("counter", Counter, 0).get()
     counter.slice(hosts=0, gpus=1).incr.broadcast()
@@ -266,8 +240,6 @@ async def test_value_mesh() -> None:
     n = proc.spawn("ctc", CastToCounter).get()
     assert list(x) == n.slice(gpus=0).doit.call_one(counter).get()
 
-    await proc.stop()
-
 
 @pytest.mark.timeout(60)
 def test_rust_binding_modules_correct() -> None:
@@ -334,8 +306,6 @@ async def test_actor_tls() -> None:
     assert 4 == await am.get.call_one()
     assert 4 == await am.get_async.call_one()
 
-    await pm.stop()
-
 
 class TLSActorFullSync(Actor):
     """An actor that manages thread-local state."""
@@ -365,8 +335,6 @@ async def test_actor_tls_full_sync() -> None:
 
     assert 4 == await am.get.call_one()
 
-    await pm.stop()
-
 
 class AsyncActor(Actor):
     def __init__(self):
@@ -393,8 +361,6 @@ async def test_async_concurrency():
     await am.no_more.call()
     await fut
 
-    await pm.stop()
-
 
 async def awaitit(f):
     return await f
@@ -751,7 +717,7 @@ async def send(self, port: Port[int]):
 
 
 @pytest.mark.timeout(60)
-async def test_port_as_argument():
+def test_port_as_argument():
     proc_mesh = local_proc_mesh(gpus=1)
     s = proc_mesh.spawn("send_alot", SendAlot).get()
     mb = Future(coro=proc_mesh._proc_mesh.task()).get().client
@@ -762,8 +728,6 @@ async def test_port_as_argument():
     for i in range(100):
         assert i == recv.recv().get()
 
-    await proc_mesh.stop()
-
 
 @pytest.mark.timeout(15)
 async def test_same_actor_twice() -> None:
@@ -780,8 +744,6 @@ async def test_same_actor_twice() -> None:
         "gspawn failed: an actor with name 'dup' has already been spawned" in error_msg
     ), f"Expected error message about duplicate actor name, got: {error_msg}"
 
-    await pm.stop()
-
 
 class TestActorMeshStop(unittest.IsolatedAsyncioTestCase):
     async def test_actor_mesh_stop(self) -> None:
@@ -808,13 +770,11 @@ def add(self, port: "Port[int]", b: int) -> None:
 
 
 @pytest.mark.timeout(60)
-async def test_ported_actor():
+def test_ported_actor():
     proc_mesh = local_proc_mesh(gpus=1).get()
     a = proc_mesh.spawn("port_actor", PortedActor).get()
     assert 5 == a.add.call_one(2).get()
 
-    await proc_mesh.stop()
-
 
 async def _recv():
     return (7, 2, 3)