Fix merge into refactored class.

canonical · Nov 27, 2024 · 8d662c4 · 8d662c4
1 parent d793be3
commit 8d662c4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/testing/src/scenario/_ops_main_mock.py b/testing/src/scenario/_ops_main_mock.py
@@ -6,7 +6,7 @@
 import marshal
 import re
 import sys
-from typing import TYPE_CHECKING, Any, Generic, Optional, Sequence, Type, cast
+from typing import TYPE_CHECKING, Any, Generic, Dict, FrozenSet, List, Sequence, Set
 
 import ops
 import ops.jujucontext
@@ -96,7 +96,7 @@ def apply_state(self, state: "State"):
             db.save_snapshot(stored_state._handle_path, stored_state.content)
 
 
-class Ops(_Manager):
+class Ops(_Manager, Generic[CharmType]):
     """Class to manage stepping through ops setup, event emission and framework commit."""
 
     def __init__(