increase health checks

integration-tests/models/test_bloom_560m.py

@@ -9,7 +9,7 @@ def bloom_560_handle(launcher):
 
 @pytest.fixture(scope="module")
 async def bloom_560(bloom_560_handle):
-    await bloom_560_handle.health(60)
+    await bloom_560_handle.health(240)
     return bloom_560_handle.client
 
 

integration-tests/models/test_bloom_560m_sharded.py

@@ -9,7 +9,7 @@ def bloom_560m_sharded_handle(launcher):
 
 @pytest.fixture(scope="module")
 async def bloom_560m_sharded(bloom_560m_sharded_handle):
-    await bloom_560m_sharded_handle.health(60)
+    await bloom_560m_sharded_handle.health(240)
     return bloom_560m_sharded_handle.client
 
 

integration-tests/models/test_flash_falcon.py

@@ -9,7 +9,7 @@ def flash_falcon_handle(launcher):
 
 @pytest.fixture(scope="module")
 async def flash_falcon(flash_falcon_handle):
-    await flash_falcon_handle.health(120)
+    await flash_falcon_handle.health(300)
     return flash_falcon_handle.client
 
 

integration-tests/models/test_flash_llama.py

@@ -9,7 +9,7 @@ def flash_llama_handle(launcher):
 
 @pytest.fixture(scope="module")
 async def flash_llama(flash_llama_handle):
-    await flash_llama_handle.health(120)
+    await flash_llama_handle.health(300)
     return flash_llama_handle.client
 
 

integration-tests/models/test_flash_neox.py

@@ -9,7 +9,7 @@ def flash_neox_handle(launcher):
 
 @pytest.fixture(scope="module")
 async def flash_neox(flash_neox_handle):
-    await flash_neox_handle.health(240)
+    await flash_neox_handle.health(300)
     return flash_neox_handle.client
 
 

integration-tests/models/test_flash_neox_sharded.py

@@ -9,7 +9,7 @@ def flash_neox_sharded_handle(launcher):
 
 @pytest.fixture(scope="module")
 async def flash_neox_sharded(flash_neox_sharded_handle):
-    await flash_neox_sharded_handle.health(240)
+    await flash_neox_sharded_handle.health(300)
     return flash_neox_sharded_handle.client
 
 

integration-tests/models/test_flash_santacoder.py

@@ -9,7 +9,7 @@ def flash_santacoder_handle(launcher):
 
 @pytest.fixture(scope="module")
 async def flash_santacoder(flash_santacoder_handle):
-    await flash_santacoder_handle.health(240)
+    await flash_santacoder_handle.health(300)
     return flash_santacoder_handle.client
 
 

integration-tests/models/test_flash_starcoder.py

@@ -9,7 +9,7 @@ def flash_starcoder_handle(launcher):
 
 @pytest.fixture(scope="module")
 async def flash_starcoder(flash_starcoder_handle):
-    await flash_starcoder_handle.health(240)
+    await flash_starcoder_handle.health(300)
     return flash_starcoder_handle.client
 
 

integration-tests/models/test_mt0_base.py

@@ -9,7 +9,7 @@ def mt0_base_handle(launcher):
 
 @pytest.fixture(scope="module")
 async def mt0_base(mt0_base_handle):
-    await mt0_base_handle.health(60)
+    await mt0_base_handle.health(300)
     return mt0_base_handle.client
 
 

integration-tests/models/test_t5_sharded.py

@@ -9,7 +9,7 @@ def t5_sharded_handle(launcher):
 
 @pytest.fixture(scope="module")
 async def t5_sharded(t5_sharded_handle):
-    await t5_sharded_handle.health(240)
+    await t5_sharded_handle.health(300)
     return t5_sharded_handle.client