From c7d0be927f7d0e1e3036f0d1edab970e15983770 Mon Sep 17 00:00:00 2001
From: Alexandru Fikl <alexfikl@gmail.com>
Date: Sun, 4 Aug 2024 09:54:42 +0300
Subject: [PATCH] fix issues flagged by mypy

---
 pytential/qbx/refinement.py | 3 +--
 test/extra_int_eq_data.py   | 2 +-
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/pytential/qbx/refinement.py b/pytential/qbx/refinement.py
index 35fde2f6b..49d1c0b36 100644
--- a/pytential/qbx/refinement.py
+++ b/pytential/qbx/refinement.py
@@ -542,9 +542,8 @@ def _visualize_refinement(actx: PyOpenCLArrayContext, discr,
 
         element_nr_base += meg.nelements
 
-    nodes_flags = DOFArray(actx, tuple(nodes_flags))
     vis_data = [
-        ("refine_flags", nodes_flags),
+        ("refine_flags", DOFArray(actx, tuple(nodes_flags))),
         ]
 
     if 0:
diff --git a/test/extra_int_eq_data.py b/test/extra_int_eq_data.py
index a6ddbf037..5d9c40260 100644
--- a/test/extra_int_eq_data.py
+++ b/test/extra_int_eq_data.py
@@ -101,7 +101,7 @@ class IntegralEquationTestCase:
     target_order: Optional[int] = None
     use_refinement: bool = True
     group_cls: Type[MeshElementGroup] = SimplexElementGroup
-    group_factory_cls: ElementGroupFactory = InterpolatoryQuadratureGroupFactory
+    group_factory_cls: Type[ElementGroupFactory] = InterpolatoryQuadratureGroupFactory
 
     # fmm
     fmm_backend: Optional[str] = "sumpy"