Minor fixes

zwicker-group · Aug 12, 2024 · d62574e · d62574e
1 parent f431191
commit d62574e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/pde/tools/plotting.py b/pde/tools/plotting.py
@@ -805,7 +805,7 @@ def in_ipython() -> bool:
 @contextlib.contextmanager
 def napari_viewer(
     grid: GridBase, run: bool | None = None, close: bool = False, **kwargs
-) -> Generator[napari.viewer.Viewer, None, None]:
+) -> Generator[napari.viewer.Viewer]:
     """Creates an napari viewer for interactive plotting.
 
     Args:

diff --git a/tests/resources/run_pde.py b/tests/resources/run_pde.py
@@ -6,10 +6,6 @@ def run_pde(t_range, storage):
     field = pde.ScalarField.random_uniform(pde.UnitGrid([8, 8]))
     storage["initial_state"] = field
 
-    # from modelrunner import model
-    #
-    # print("PATH", model.__file__)
-
     eq = pde.DiffusionPDE()
     result = eq.solve(
         field,