Partial revert of D75106997 (#31)

Summary:
Pull Request resolved: #31

D75106997 appears to have broken github CI, and this partial revert doesn't seem to reintroduce any warnings, so hopefully this fixes things.

Reviewed By: colin2328

Differential Revision: D75314370

fbshipit-source-id: 16caf12df35e53a20b547d4d9b5c290ea67a4641

Author: Sam Lurye

python/tests/test_controller.py

@@ -646,11 +646,10 @@ def test_panicking_worker():
 
 def test_timeout_warning(caplog):
     timeout = 3
-    with local.local_device_mesh(
+    with local_rust_device_mesh(
         1,
         2,
         True,
-        rust=True,
         controller_params=ControllerParams(1, timeout, 100, False),
     ) as dm:
         for _ in range(3):
@@ -676,11 +675,10 @@ def test_timeout_warning(caplog):
 
 def test_timeout_failure():
     timeout = 3
-    with local.local_device_mesh(
+    with local_rust_device_mesh(
         1,
         1,
         True,
-        rust=True,
         controller_params=ControllerParams(1, timeout, 100, True),
     ) as dm:
         for _ in range(3):