Misc CI fixes (#3780)

* Fix flaky FS test

* Remove unused file

* Exclude vllm integration to install more recent torch

* Log CLI command output if failed

* Unused ignore

* Load iris dataset using sklearn

Author: schustmi

.test_durations

@@ -36,7 +36,7 @@ install_integrations() {
     # figure out the python version
     python_version=$(python -c "import sys; print('.'.join(map(str, sys.version_info[:2])))")
 
-    ignore_integrations="feast label_studio bentoml seldon pycaret skypilot_aws skypilot_gcp skypilot_azure skypilot_kubernetes skypilot_lambda pigeon prodigy argilla"
+    ignore_integrations="feast label_studio bentoml seldon pycaret skypilot_aws skypilot_gcp skypilot_azure skypilot_kubernetes skypilot_lambda pigeon prodigy argilla vllm"
 
     # Ignore tensorflow and deepchecks only on Python 3.12
     if [ "$python_version" = "3.12" ]; then

scripts/install-zenml-dev.sh

@@ -171,7 +171,7 @@ for COMMAND in "${COMMANDS[@]}"; do
     if [ -n "$TIMEOUT_CMD" ]; then
       # Use set +e to prevent script from exiting if the command times out or fails
       set +e
-      $TIMEOUT_CMD $TIMEOUT_SECONDS bash -c "$COMMAND" > /dev/null 2>&1
+      COMMAND_OUTPUT=$($TIMEOUT_CMD $TIMEOUT_SECONDS bash -c "$COMMAND" 2>&1)
       EXIT_CODE=$?
       # Restore error handling
       set -e
@@ -186,13 +186,14 @@ for COMMAND in "${COMMANDS[@]}"; do
         COMMAND_FAILED=true
         ERROR_MESSAGE="Command failed on run $i (exit code: $EXIT_CODE)"
         echo "❌ $ERROR_MESSAGE"
+        echo "$COMMAND_OUTPUT"
         break
       fi
     else
       # No timeout command available, just run normally
       # Use set +e to prevent script from exiting if the command fails
       set +e
-      eval $COMMAND > /dev/null 2>&1
+      COMMAND_OUTPUT=$(eval $COMMAND 2>&1)
       EXIT_CODE=$?
       # Restore error handling
       set -e
@@ -201,6 +202,7 @@ for COMMAND in "${COMMANDS[@]}"; do
         COMMAND_FAILED=true
         ERROR_MESSAGE="Command failed on run $i (exit code: $EXIT_CODE)"
         echo "❌ $ERROR_MESSAGE"
+        echo "$COMMAND_OUTPUT"
         break
       fi
     fi

scripts/profile-cli.sh

@@ -150,7 +150,7 @@ def run(self) -> None:
 
         try:
             parser: argparse.ArgumentParser = make_arg_parser(
-                FlexibleArgumentParser()  # type: ignore[no-untyped-call]
+                FlexibleArgumentParser()
             )
             # pass in empty list to get default args
             # otherwise it will try to get the args from sys.argv

src/zenml/integrations/vllm/services/vllm_deployment.py

@@ -14,7 +14,7 @@
 from typing import Tuple
 
 import pandas as pd
-from deepchecks.tabular.datasets.classification import iris
+from sklearn.datasets import load_iris
 from sklearn.model_selection import train_test_split
 from typing_extensions import Annotated
 
@@ -27,9 +27,8 @@ def data_loader() -> Tuple[
     Annotated[pd.DataFrame, "comparison_dataset"],
 ]:
     """Load the iris dataset."""
-    iris_df = iris.load_data(data_format="Dataframe", as_train_test=False)
-    label_col = "target"
+    iris = load_iris(as_frame=True)
     df_train, df_test = train_test_split(
-        iris_df, stratify=iris_df[label_col], random_state=0
+        iris["frame"], stratify=iris["target"], random_state=0
     )
     return df_train, df_test

tests/integration/examples/deepchecks/steps/data_loader.py

@@ -212,8 +212,13 @@ def test_is_remote_when_using_remote_prefix(filesystem):
 def test_is_remote_when_using_non_remote_prefix(filesystem):
     """is_remote returns False when path doesn't start with
     a remote prefix"""
+    expected_result = False
+
+    if filesystem in REMOTE_FS_PREFIX:
+        expected_result = True
+
     some_random_path = os.path.join(f"{filesystem}some_directory")
-    assert io_utils.is_remote(some_random_path) is False
+    assert io_utils.is_remote(some_random_path) is expected_result
 
 
 def test_create_file_if_not_exists(tmp_path) -> None: