diff --git a/poetry.lock b/poetry.lock
index a9db5ff67..45756ec3e 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1,4 +1,4 @@
-# This file is automatically @generated by Poetry 1.4.2 and should not be changed by hand.
+# This file is automatically @generated by Poetry and should not be changed by hand.
 
 [[package]]
 name = "aiohttp"
@@ -428,13 +428,13 @@ colorama = {version = "*", markers = "platform_system == \"Windows\""}
 
 [[package]]
 name = "codeflare-torchx"
-version = "0.6.0.dev1"
+version = "0.6.0.dev2"
 description = "TorchX SDK and Components"
 category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
-    {file = "codeflare_torchx-0.6.0.dev1-py3-none-any.whl", hash = "sha256:a03bf570c775cb7fc7787a436ef5a76c726669e287af510513c5bc03033d1fca"},
+    {file = "codeflare_torchx-0.6.0.dev2-py3-none-any.whl", hash = "sha256:d303efffb9b1e105390ed672a3358de40174146530929df83c7d7af27372fbcc"},
 ]
 
 [package.dependencies]
@@ -2149,8 +2149,8 @@ frozenlist = "*"
 fsspec = {version = "*", optional = true, markers = "extra == \"data\""}
 gpustat = {version = ">=1.0.0", optional = true, markers = "extra == \"default\""}
 grpcio = [
-    {version = ">=1.32.0", optional = true, markers = "python_version < \"3.10\" and extra == \"default\""},
-    {version = ">=1.42.0", optional = true, markers = "python_version >= \"3.10\" and extra == \"default\""},
+    {version = ">=1.32.0", optional = true, markers = "python_version < \"3.10\""},
+    {version = ">=1.42.0", optional = true, markers = "python_version >= \"3.10\""},
 ]
 jsonschema = "*"
 msgpack = ">=1.0.0,<2.0.0"
@@ -2695,4 +2695,4 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.8"
-content-hash = "e7fa79bd035b5bffcb1668c0a8cf8fa4e869a614b934a09100ab1d53338fe11b"
+content-hash = "33f4e77105de697fd6fc15537180df77ab057e3d2e5db27d7a9a2f8cc9f69a03"
diff --git a/pyproject.toml b/pyproject.toml
index e262b5270..953ce0f69 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,7 +25,7 @@ openshift-client = "1.0.18"
 rich = "^12.5"
 ray = {version = "2.7.0", extras = ["data", "default"]}
 kubernetes = ">= 25.3.0, < 27"
-codeflare-torchx = "0.6.0.dev1"
+codeflare-torchx = "0.6.0.dev2"
 cryptography = "40.0.2"
 executing = "1.2.0"
 pydantic = "< 2"