@@ -146,11 +146,28 @@ jobs:
146146 - uses : ./.github/actions/setup-java
147147 with :
148148 java-version : ${{ matrix.java }}
149+ # TODO: Remove this step once we remove the pyspark constraint in requirements/constraints.txt
150+ - name : Remove pyspark constraint
151+ run : |
152+ if [ "${{ matrix.package }}" = "pyspark" ]; then
153+ sed -i '/^pyspark/d' requirements/constraints.txt
154+ git config user.name 'mlflow-app[bot]'
155+ git config user.email 'mlflow-app[bot]@users.noreply.github.com'
156+ git add requirements/constraints.txt
157+ git commit -m "Remove pyspark constraints for testing"
158+ fi
149159 - name : Install mlflow & test dependencies
150160 run : |
151161 pip install -U pip wheel setuptools
152- pip install .[extras]
153- pip install -r requirements/test-requirements.txt
162+ # For tracing SDK test, install the tracing package from the local path and minimal test dependencies
163+ if [[ "${{ matrix.category }}" == "tracing-sdk" ]]; then
164+ pip install packages/tracing
165+ pip install pytest pytest-asyncio pytest-cov
166+ # Other two categories of tests (model/autologging)
167+ else
168+ pip install .[extras]
169+ pip install -r requirements/test-requirements.txt
170+ fi
154171 - name : Install ${{ matrix.package }} ${{ matrix.version }}
155172 run : |
156173 ${{ matrix.install }}
@@ -193,11 +210,28 @@ jobs:
193210 - uses : ./.github/actions/setup-java
194211 with :
195212 java-version : ${{ matrix.java }}
213+ # TODO: Remove this step once we remove the pyspark constraint in requirements/constraints.txt
214+ - name : Remove pyspark constraint
215+ run : |
216+ if [ "${{ matrix.package }}" = "pyspark" ]; then
217+ sed -i '/^pyspark/d' requirements/constraints.txt
218+ git config user.name 'mlflow-app[bot]'
219+ git config user.email 'mlflow-app[bot]@users.noreply.github.com'
220+ git add requirements/constraints.txt
221+ git commit -m "Remove pyspark constraints for testing"
222+ fi
196223 - name : Install mlflow & test dependencies
197224 run : |
198225 pip install -U pip wheel setuptools
199- pip install .[extras]
200- pip install -r requirements/test-requirements.txt
226+ # For tracing SDK test, install the tracing package from the local path and minimal test dependencies
227+ if [[ "${{ matrix.category }}" == "tracing-sdk" ]]; then
228+ pip install packages/tracing
229+ pip install pytest pytest-asyncio pytest-cov
230+ # Other two categories of tests (model/autologging)
231+ else
232+ pip install .[extras]
233+ pip install -r requirements/test-requirements.txt
234+ fi
201235 - name : Install ${{ matrix.package }} ${{ matrix.version }}
202236 run : |
203237 ${{ matrix.install }}
0 commit comments