Fix tests

canonical · Apr 7, 2023 · 9d5422b · 9d5422b
1 parent 6e54c60
commit 9d5422b
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/charm.py b/src/charm.py
@@ -10,7 +10,7 @@
 from functools import cached_property
 from os.path import join
 from pathlib import Path
-from typing import Any, List, Optional
+from typing import TYPE_CHECKING, Any, List, Optional
 
 from charms.data_platform_libs.v0.data_interfaces import (
     DatabaseCreatedEvent,
@@ -36,7 +36,12 @@
 from ops.charm import CharmBase, ConfigChangedEvent, HookEvent, PebbleReadyEvent, RelationEvent
 from ops.main import main
 from ops.model import ActiveStatus, BlockedStatus, MaintenanceStatus, ModelError, WaitingStatus
-from ops.pebble import ExecError, Layer, LayerDict
+from ops.pebble import ExecError, Layer
+
+
+if TYPE_CHECKING:
+    from ops.pebble import LayerDict
+
 
 logger = logging.getLogger(__name__)
 KRATOS_ADMIN_PORT = 4434