Remove v1 references (#61)

* remove v0 references

* flake

Author: LucasLLC

tests/test_models.py

@@ -4,16 +4,6 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-from torchstore.constants import MONARCH_HOSTMESH_V1
-
-if MONARCH_HOSTMESH_V1:
-    from monarch._rust_bindings.monarch_hyperactor.channel import ChannelTransport
-    from monarch._rust_bindings.monarch_hyperactor.config import configure
-
-    configure(
-        default_transport=ChannelTransport.MetaTlsWithHostname,
-    )
-
 import math
 import os
 import tempfile

torchstore/api.py

@@ -8,10 +8,11 @@
 
 import torch
 
+from monarch.actor import get_or_spawn_controller
+
 import torchstore.state_dict_utils
 from torchstore.client import LocalClient
 
-from torchstore.constants import MONARCH_HOSTMESH_V1
 from torchstore.controller import Controller
 from torchstore.storage_volume import StorageVolume
 from torchstore.strategy import (
@@ -21,11 +22,6 @@
 )
 from torchstore.transport.pipe import TensorSlice
 
-if MONARCH_HOSTMESH_V1:
-    from monarch._src.actor.v1.proc_mesh import get_or_spawn_controller
-else:
-    from monarch.actor import get_or_spawn_controller
-
 
 # I need to keep this somewhere, so here we go
 DEFAULT_TORCHSTORE_NAME: str = "TorchStore"

torchstore/constants.py

@@ -11,12 +11,7 @@
 
 import torch
 
-from torchstore.constants import MONARCH_HOSTMESH_V1
-
-if MONARCH_HOSTMESH_V1:
-    from monarch._src.actor.v1.host_mesh import this_host
-else:
-    from monarch.actor import this_host
+from monarch.actor import this_host
 
 from torchstore.transport import TensorSlice