Fix comments

Ostorlab · Dec 3, 2024 · 2517c31 · 2517c31
1 parent 13a4992
commit 2517c31
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/ostorlab/runtimes/local/runtime.py b/src/ostorlab/runtimes/local/runtime.py
@@ -311,7 +311,7 @@ def stop(self, scan_id: str) -> None:
             network_labels = network.attrs["Labels"]
             if (
                 network_labels is not None
-                and network_labels.get("ostorlab.universe") == scan_id
+                and int(network_labels.get("ostorlab.universe")) == scan_id
             ):
                 logger.info("removing network %s", network_labels)
                 stopped_network.append(network)
@@ -320,7 +320,10 @@ def stop(self, scan_id: str) -> None:
         configs = self._docker_client.configs.list()
         for config in configs:
             config_labels = config.attrs["Spec"]["Labels"]
-            if config_labels.get("ostorlab.universe") == scan_id:
+            if (
+                config_labels.get("ostorlab.universe") is not None
+                and int(config_labels.get("ostorlab.universe")) == scan_id
+            ):
                 logger.info("removing config %s", config_labels)
                 stopped_configs.append(config)
                 config.remove()