From d020e7ec960da8d582ec74ea63a72356ae561f1f Mon Sep 17 00:00:00 2001
From: Victor Turrisi <vt.turrisi@gmail.com>
Date: Tue, 12 Oct 2021 18:49:34 -0300
Subject: [PATCH] fixed tests

---
 tests/args/test_setup.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/tests/args/test_setup.py b/tests/args/test_setup.py
index de65f04b..5ba2525c 100644
--- a/tests/args/test_setup.py
+++ b/tests/args/test_setup.py
@@ -204,6 +204,9 @@ def test_additional_setup_pretrain():
         "contrast": [0.4],
         "saturation": [0.2],
         "hue": [0.1],
+        "color_jitter_prob": [0.8],
+        "gray_scale_prob": [0.2],
+        "horizontal_flip_prob": [0.5],
         "gaussian_prob": [1.0, 0.1],
         "solarization_prob": [0.2, 0.1],
         "min_scale": [0.08],
@@ -234,6 +237,9 @@ def test_additional_setup_pretrain():
         "contrast": [0.4],
         "saturation": [0.2],
         "hue": [0.1],
+        "color_jitter_prob": [0.8],
+        "gray_scale_prob": [0.2],
+        "horizontal_flip_prob": [0.5],
         "gaussian_prob": [0.5],
         "solarization_prob": [0.5],
         "min_scale": [0.08],
@@ -264,6 +270,9 @@ def test_additional_setup_pretrain():
         "contrast": [0.4],
         "saturation": [0.2],
         "hue": [0.1],
+        "color_jitter_prob": [0.8],
+        "gray_scale_prob": [0.2],
+        "horizontal_flip_prob": [0.5],
         "gaussian_prob": [0.5],
         "solarization_prob": [0.5],
         "min_scale": [0.08],
@@ -294,6 +303,9 @@ def test_additional_setup_pretrain():
         "contrast": [0.4],
         "saturation": [0.2],
         "hue": [0.1],
+        "color_jitter_prob": [0.8],
+        "gray_scale_prob": [0.2],
+        "horizontal_flip_prob": [0.5],
         "gaussian_prob": [0.5, 0.2],
         "solarization_prob": [0.5, 0.3],
         "min_scale": [0.08],