bug fix

kipoi · Apr 4, 2022 · 54c3182 · 54c3182
1 parent 76c0ec1
commit 54c3182
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test-containers/test_containers_from_command_line.py b/test-containers/test_containers_from_command_line.py
@@ -40,9 +40,9 @@ def test_images(self, test_docker_image):
                     if self.modelgroup_name != "DeepSEA":
                         break
         elif self.image_name not in [None, "kipoi-base-env"]:
-            all_models = self.docker_to_model_dict.get(self.image_name)
+            models = self.docker_to_model_dict.get(self.image_name)
             if "shared" in self.image_name:
-                models = one_model_per_modelgroup(all_models)
+                models = one_model_per_modelgroup(models)
             for model in models:
                 print(f"Testing {model} with {self.image_name}")
                 test_docker_image(model_name=model, image_name=self.image_name)