From 66b2d3f582ab6be37a11447d099c9cbde059b913 Mon Sep 17 00:00:00 2001
From: Oskar Taubert <oskar.taubert@kit.edu>
Date: Fri, 19 Apr 2024 14:16:07 +0200
Subject: [PATCH] reduced the maximum network depth for testing

---
 tests/test_surrogate.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_surrogate.py b/tests/test_surrogate.py
index 9f5b2fa1..5d35fd75 100644
--- a/tests/test_surrogate.py
+++ b/tests/test_surrogate.py
@@ -312,7 +312,7 @@ def test_mnist_static(mpi_tmp_path):
     num_generations = 3  # Number of generations
     pop_size = 2 * MPI.COMM_WORLD.size  # Breeding population size
     limits = {
-        "conv_layers": (2, 10),
+        "conv_layers": (2, 3),
         "activation": ("relu", "sigmoid", "tanh"),
         "lr": (0.01, 0.0001),
     }  # Define search space.
@@ -350,7 +350,7 @@ def test_mnist_dynamic(mpi_tmp_path):
     num_generations = 3  # Number of generations
     pop_size = 2 * MPI.COMM_WORLD.size  # Breeding population size
     limits = {
-        "conv_layers": (2, 10),
+        "conv_layers": (2, 3),
         "activation": ("relu", "sigmoid", "tanh"),
         "lr": (0.01, 0.0001),
     }  # Define search space.