Fix MLFlow test (#2161)

Author: avishniakov

.github/workflows/integration-test.yml

@@ -188,7 +188,7 @@ jobs:
         run: pip install docker-compose
 
       - name: Install Linux System Dependencies
-        if: runner.os=='Linux'
+        if: inputs.os=='ubuntu-latest'
         run: sudo apt install graphviz
 
       - name: Install MacOS System Dependencies
@@ -219,7 +219,7 @@ jobs:
         run: |
           curl -LO "https://dl.k8s.io/release/$(curl -L -s https://dl.k8s.io/release/stable.txt)/bin/linux/amd64/kubectl"
           sudo install -o root -g 0 -m 0755 kubectl /usr/local/bin/kubectl
-        if: runner.os=='Linux'
+        if: inputs.os=='ubuntu-latest'
 
       - name: Install kubectl on MacOS
         run: |

tests/integration/examples/mlflow/pipelines/deployment_pipelines/deployment_inference_pipeline.py

@@ -23,7 +23,7 @@
 docker_settings = DockerSettings(required_integrations=[MLFLOW, TENSORFLOW])
 
 
-@pipeline(enable_cache=True, settings={"docker": docker_settings})
+@pipeline(enable_cache=False, settings={"docker": docker_settings})
 def mlflow_deployment_inference_pipeline(
     pipeline_name: str = "mlflow_train_deploy_pipeline",
     pipeline_step_name: str = "mlflow_model_deployer_step",

tests/integration/examples/mlflow/pipelines/deployment_pipelines/deployment_training_pipeline.py

@@ -25,7 +25,7 @@
 docker_settings = DockerSettings(required_integrations=[MLFLOW, TENSORFLOW])
 
 
-@pipeline(enable_cache=True, settings={"docker": docker_settings})
+@pipeline(enable_cache=False, settings={"docker": docker_settings})
 def mlflow_train_deploy_pipeline(
     epochs: int = 1,
     lr: float = 0.001,