Addresses a bug in padding with random mutations

Author: miguelgondu

src/poli_baselines/core/utils/mutations/random_mutations.py

@@ -25,7 +25,14 @@ def add_random_mutations_to_reach_pop_size(
             f"Expected x to have less elements than the population size. x.shape: {x.shape}, pop_size: {population_size}"
         )
 
-    x_as_1d_array_of_strings = np.array(["".join(x[i, :]) for i in range(b)])
+    if len(x.shape) == 1:
+        # Then the input as an array [b,] of strings
+        x_as_1d_array_of_strings = x
+    elif len(x.shape) == 2:
+        # Then the input as an array [b, L] of strings
+        x_as_1d_array_of_strings = np.array(["".join(x[i, :]) for i in range(b)])
+    else:
+        raise ValueError(f"Expected x to have shape [b,] or [b, L], but got {x.shape}.")
 
     more_mutants = []
     for _ in range(population_size - b):

src/poli_baselines/tests/solvers/bayesian_optimization/test_lambo2.py

@@ -1,19 +1,17 @@
-import importlib
 from pathlib import Path
 
-import pytest
-from poli.repository import EhrlichProblemFactory
+from poli.repository import EhrlichHoloProblemFactory
 
-if importlib.util.find_spec("cortex") is None:
-    pytest.skip("Cortex is not installed.", allow_module_level=True)
+# if importlib.util.find_spec("cortex") is None:
+#     pytest.skip("Cortex is not installed.", allow_module_level=True)
 
 TEST_ASSETS = Path(__file__).parent.parent.parent / "test_files"
 
 
 def test_lambo2_runs_on_ehrlich():
     from poli_baselines.solvers.bayesian_optimization.lambo2 import LaMBO2
 
-    problem = EhrlichProblemFactory().create(
+    problem = EhrlichHoloProblemFactory().create(
         sequence_length=10,
         n_motifs=2,
         motif_length=4,