prompt: We are currently solving the following issue within our repository. Here is the issue text:
  --- BEGIN ISSUE ---
  [FEAT][CV] Add conditions to checks missing conditions
  Some checks are missing conditions:
  
  - [x] Heatmap
  - [x] Image Drift
  - [x] Train Test Drift
  - [x] Robustness 
  
  --- END ISSUE ---
  
  Below are some code segments, each from a relevant file. One or more of these files may contain bugs.
  --- BEGIN FILES ---
  Path: `deepchecks/vision/suites/default_suites.py`
  Content:
  ```
  1 # ----------------------------------------------------------------------------
  2 # Copyright (C) 2021-2022 Deepchecks (https://www.deepchecks.com)
  3 #
  4 # This file is part of Deepchecks.
  5 # Deepchecks is distributed under the terms of the GNU Affero General
  6 # Public License (version 3 or later).
  7 # You should have received a copy of the GNU Affero General Public License
  8 # along with Deepchecks.  If not, see <http://www.gnu.org/licenses/>.
  9 # ----------------------------------------------------------------------------
  10 #
  11 """Functions for loading the default (built-in) vision suites for various validation stages.
  12 
  13 Each function returns a new suite that is initialized with a list of checks and default conditions.
  14 It is possible to customize these suites by editing the checks and conditions inside it after the suites' creation.
  15 """
  16 from deepchecks.vision.checks import ClassPerformance, TrainTestLabelDrift, MeanAveragePrecisionReport, \
  17     MeanAverageRecallReport, ImagePropertyDrift, ImageDatasetDrift, SimpleModelComparison, ConfusionMatrixReport, \
  18     RobustnessReport, TrainTestPredictionDrift
  19 from deepchecks.vision import Suite
  20 
  21 
  22 __all__ = ['train_test_validation', 'model_evaluation', 'full_suite']
  23 
  24 from deepchecks.vision.checks.distribution import HeatmapComparison
  25 
  26 
  27 def train_test_validation() -> Suite:
  28     """Create a suite that is meant to validate correctness of train-test split, including integrity, \
  29     distribution and leakage checks."""
  30     return Suite(
  31         'Train Test Validation Suite',
  32         HeatmapComparison(),
  33         TrainTestLabelDrift(),
  34         TrainTestPredictionDrift(),
  35         ImagePropertyDrift().add_condition_drift_score_not_greater_than(),
  36         ImageDatasetDrift()
  37     )
  38 
  39 
  40 def model_evaluation() -> Suite:
  41     """Create a suite that is meant to test model performance and overfit."""
  42     return Suite(
  43         'Model Evaluation Suite',
  44         ClassPerformance(),
  45         MeanAveragePrecisionReport(),
  46         MeanAverageRecallReport(),
  47         SimpleModelComparison(),
  48         ConfusionMatrixReport(),
  49         RobustnessReport().add_condition_degradation_not_greater_than()
  50     )
  51 
  52 
  53 def full_suite() -> Suite:
  54     """Create a suite that includes many of the implemented checks, for a quick overview of your model and data."""
  55     return Suite(
  56         'Full Suite',
  57         model_evaluation(),
  58         train_test_validation(),
  59     )
  60
  ```
  --- END FILES ---
  
  Please first localize the bug based on the issue statement, and then generate a patch according to the `git diff` format fenced by three backticks.
  
  Here is an example:
  
  ```diff
  diff --git a/examples/server_async.py b/examples/server_async.py
  --- a/examples/server_async.py
  +++ b/examples/server_async.py
  @@ -313,4 +313,4 @@
   
   
   if __name__ == "__main__":
  -    asyncio.run(run_async_server("."), debug=True)
  +    asyncio.run(run_async_server(), debug=True)
  diff --git a/examples/server_sync.py b/examples/server_sync.py
  --- a/examples/server_sync.py
  +++ b/examples/server_sync.py
  @@ -313,5 +313,5 @@
   
   
   if __name__ == "__main__":
  -    server = run_sync_server(".")
  +    server = run_sync_server()
       server.shutdown()
  
  ```
  
golden_diff: diff --git a/deepchecks/vision/suites/default_suites.py b/deepchecks/vision/suites/default_suites.py
  --- a/deepchecks/vision/suites/default_suites.py
  +++ b/deepchecks/vision/suites/default_suites.py
  @@ -31,7 +31,7 @@
           'Train Test Validation Suite',
           HeatmapComparison(),
           TrainTestLabelDrift(),
  -        TrainTestPredictionDrift(),
  +        TrainTestPredictionDrift().add_condition_drift_score_not_greater_than(),
           ImagePropertyDrift().add_condition_drift_score_not_greater_than(),
           ImageDatasetDrift()
       )
  
verification_info: {"golden_diff": "diff --git a/deepchecks/vision/suites/default_suites.py b/deepchecks/vision/suites/default_suites.py\n--- a/deepchecks/vision/suites/default_suites.py\n+++ b/deepchecks/vision/suites/default_suites.py\n@@ -31,7 +31,7 @@\n         'Train Test Validation Suite',\n         HeatmapComparison(),\n         TrainTestLabelDrift(),\n-        TrainTestPredictionDrift(),\n+        TrainTestPredictionDrift().add_condition_drift_score_not_greater_than(),\n         ImagePropertyDrift().add_condition_drift_score_not_greater_than(),\n         ImageDatasetDrift()\n     )\n", "issue": "[FEAT][CV] Add conditions to checks missing conditions\nSome checks are missing conditions:\r\n\r\n- [x] Heatmap\r\n- [x] Image Drift\r\n- [x] Train Test Drift\r\n- [x] Robustness \n", "before_files": [{"content": "# ----------------------------------------------------------------------------\n# Copyright (C) 2021-2022 Deepchecks (https://www.deepchecks.com)\n#\n# This file is part of Deepchecks.\n# Deepchecks is distributed under the terms of the GNU Affero General\n# Public License (version 3 or later).\n# You should have received a copy of the GNU Affero General Public License\n# along with Deepchecks.  If not, see <http://www.gnu.org/licenses/>.\n# ----------------------------------------------------------------------------\n#\n\"\"\"Functions for loading the default (built-in) vision suites for various validation stages.\n\nEach function returns a new suite that is initialized with a list of checks and default conditions.\nIt is possible to customize these suites by editing the checks and conditions inside it after the suites' creation.\n\"\"\"\nfrom deepchecks.vision.checks import ClassPerformance, TrainTestLabelDrift, MeanAveragePrecisionReport, \\\n    MeanAverageRecallReport, ImagePropertyDrift, ImageDatasetDrift, SimpleModelComparison, ConfusionMatrixReport, \\\n    RobustnessReport, TrainTestPredictionDrift\nfrom deepchecks.vision import Suite\n\n\n__all__ = ['train_test_validation', 'model_evaluation', 'full_suite']\n\nfrom deepchecks.vision.checks.distribution import HeatmapComparison\n\n\ndef train_test_validation() -> Suite:\n    \"\"\"Create a suite that is meant to validate correctness of train-test split, including integrity, \\\n    distribution and leakage checks.\"\"\"\n    return Suite(\n        'Train Test Validation Suite',\n        HeatmapComparison(),\n        TrainTestLabelDrift(),\n        TrainTestPredictionDrift(),\n        ImagePropertyDrift().add_condition_drift_score_not_greater_than(),\n        ImageDatasetDrift()\n    )\n\n\ndef model_evaluation() -> Suite:\n    \"\"\"Create a suite that is meant to test model performance and overfit.\"\"\"\n    return Suite(\n        'Model Evaluation Suite',\n        ClassPerformance(),\n        MeanAveragePrecisionReport(),\n        MeanAverageRecallReport(),\n        SimpleModelComparison(),\n        ConfusionMatrixReport(),\n        RobustnessReport().add_condition_degradation_not_greater_than()\n    )\n\n\ndef full_suite() -> Suite:\n    \"\"\"Create a suite that includes many of the implemented checks, for a quick overview of your model and data.\"\"\"\n    return Suite(\n        'Full Suite',\n        model_evaluation(),\n        train_test_validation(),\n    )\n", "path": "deepchecks/vision/suites/default_suites.py"}], "after_files": [{"content": "# ----------------------------------------------------------------------------\n# Copyright (C) 2021-2022 Deepchecks (https://www.deepchecks.com)\n#\n# This file is part of Deepchecks.\n# Deepchecks is distributed under the terms of the GNU Affero General\n# Public License (version 3 or later).\n# You should have received a copy of the GNU Affero General Public License\n# along with Deepchecks.  If not, see <http://www.gnu.org/licenses/>.\n# ----------------------------------------------------------------------------\n#\n\"\"\"Functions for loading the default (built-in) vision suites for various validation stages.\n\nEach function returns a new suite that is initialized with a list of checks and default conditions.\nIt is possible to customize these suites by editing the checks and conditions inside it after the suites' creation.\n\"\"\"\nfrom deepchecks.vision.checks import ClassPerformance, TrainTestLabelDrift, MeanAveragePrecisionReport, \\\n    MeanAverageRecallReport, ImagePropertyDrift, ImageDatasetDrift, SimpleModelComparison, ConfusionMatrixReport, \\\n    RobustnessReport, TrainTestPredictionDrift\nfrom deepchecks.vision import Suite\n\n\n__all__ = ['train_test_validation', 'model_evaluation', 'full_suite']\n\nfrom deepchecks.vision.checks.distribution import HeatmapComparison\n\n\ndef train_test_validation() -> Suite:\n    \"\"\"Create a suite that is meant to validate correctness of train-test split, including integrity, \\\n    distribution and leakage checks.\"\"\"\n    return Suite(\n        'Train Test Validation Suite',\n        HeatmapComparison(),\n        TrainTestLabelDrift(),\n        TrainTestPredictionDrift().add_condition_drift_score_not_greater_than(),\n        ImagePropertyDrift().add_condition_drift_score_not_greater_than(),\n        ImageDatasetDrift()\n    )\n\n\ndef model_evaluation() -> Suite:\n    \"\"\"Create a suite that is meant to test model performance and overfit.\"\"\"\n    return Suite(\n        'Model Evaluation Suite',\n        ClassPerformance(),\n        MeanAveragePrecisionReport(),\n        MeanAverageRecallReport(),\n        SimpleModelComparison(),\n        ConfusionMatrixReport(),\n        RobustnessReport().add_condition_degradation_not_greater_than()\n    )\n\n\ndef full_suite() -> Suite:\n    \"\"\"Create a suite that includes many of the implemented checks, for a quick overview of your model and data.\"\"\"\n    return Suite(\n        'Full Suite',\n        model_evaluation(),\n        train_test_validation(),\n    )\n", "path": "deepchecks/vision/suites/default_suites.py"}]}