remove type hints

ahalev · Mar 31, 2023 · 16fdf58 · 16fdf58
1 parent e8784db
commit 16fdf58
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/pymgrid/utils/dry_run.py b/src/pymgrid/utils/dry_run.py
@@ -1,13 +1,10 @@
 import yaml
 
 from contextlib import contextmanager
-from typing import Union
-
-import pymgrid
 
 
 @contextmanager
-def dry_run(pymgrid_object: Union['pymgrid.Microgrid', 'pymgrid.modules.base.BaseMicrogridModule']):
+def dry_run(pymgrid_object):
     """
     A context manager to test modifications of a pymgrid object without modifying said object.