modify cluster status checks tests (#1197)

Co-authored-by: Alexandra Belousov <sashabelousovrh@Alexandras-MacBook-Pro.local>
run-house · Sep 17, 2024 · 4569ea2 · 4569ea2
1 parent a742de3
commit 4569ea2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/runhouse/servers/cluster_servlet.py b/runhouse/servers/cluster_servlet.py
@@ -93,6 +93,7 @@ async def __init__(
             )
             collect_gpu_thread.start()
 
+
         if self.cluster_config.get("has_cuda"):
             logger.debug("Creating _periodic_gpu_check thread.")
             collect_gpu_thread = threading.Thread(

diff --git a/tests/test_servers/test_servlet.py b/tests/test_servers/test_servlet.py
@@ -23,10 +23,10 @@ def test_put_resource(self, test_env_servlet):
         assert deserialize_data(resp.data, resp.serialization) == resource.name
 
     @pytest.mark.level("unit")
-    def test_put_obj_local(self, test_servlet):
+    def test_put_obj_local(self, test_env_servlet):
         resource = Resource(name="local-resource")
         resp = ObjStore.call_actor_method(
-            test_servlet,
+            test_env_servlet,
             "aput_local",
             key="key1",
             data=serialize_data(resource, "pickle"),
@@ -78,6 +78,7 @@ def test_get_obj(self, test_env_servlet):
     def test_get_obj_remote(self, test_env_servlet):
         resp = ObjStore.call_actor_method(
             test_env_servlet,
+
             "aget_local",
             key="key1",
             default=KeyError,