From 341a8bcb0d78c5f7a0346f92189b44587e9debf2 Mon Sep 17 00:00:00 2001
From: Alex Sherstinsky <alexsherstinsky@users.noreply.github.com>
Date: Tue, 30 Jul 2024 10:28:36 -0700
Subject: [PATCH] Update the trunk version to "0.10.4.dev" (#4021)

---
 .github/workflows/pytest.yml | 3 ++-
 ludwig/globals.py            | 2 +-
 requirements_test.txt        | 2 +-
 setup.py                     | 2 +-
 4 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
index 4f90d4b138c..03a307da140 100644
--- a/.github/workflows/pytest.yml
+++ b/.github/workflows/pytest.yml
@@ -348,6 +348,7 @@ jobs:
           python --version
           pip --version
           python -m pip install -U pip
+          pip install torch==2.0.0 torchtext torchvision torchaudio
           pip install '.[test]'
           pip list
         shell: bash
@@ -387,9 +388,9 @@ jobs:
           python --version
           pip --version
           python -m pip install -U pip
-          pip install torch==2.0.0 torchtext
           pip install ray==2.3.0
           pip install '.'
+          pip install torch==2.0.0 torchtext torchvision torchaudio
           pip list
         shell: bash
       - name: Check Install
diff --git a/ludwig/globals.py b/ludwig/globals.py
index 2e8c8f82217..f43ec6e366c 100644
--- a/ludwig/globals.py
+++ b/ludwig/globals.py
@@ -14,7 +14,7 @@
 # limitations under the License.
 # ==============================================================================
 
-LUDWIG_VERSION = "0.10.3.dev"
+LUDWIG_VERSION = "0.10.4.dev"
 
 MODEL_FILE_NAME = "model"
 MODEL_WEIGHTS_FILE_NAME = "model_weights"
diff --git a/requirements_test.txt b/requirements_test.txt
index 84479ab0497..f42f76db6a7 100644
--- a/requirements_test.txt
+++ b/requirements_test.txt
@@ -11,7 +11,7 @@ mlflow
 # For testing optional Ray Tune Search Algorithms
 # search_alg: bohb
 hpbandster
-ConfigSpace
+ConfigSpace==0.7.1
 
 # search_alg: ax
 ax-platform
diff --git a/setup.py b/setup.py
index f285107fa76..7276b5fabf7 100644
--- a/setup.py
+++ b/setup.py
@@ -49,7 +49,7 @@
 
 setup(
     name="ludwig",
-    version="0.10.3.dev",
+    version="0.10.4.dev",
     description="Declarative machine learning: End-to-end machine learning pipelines using data-driven configurations.",
     long_description=long_description,
     long_description_content_type="text/markdown",