fix incorrect dataclass freeze

inducer · Sep 23, 2022 · 6668ccc · 6668ccc
1 parent 16740c8
commit 6668ccc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/pytential/qbx/geometry.py b/pytential/qbx/geometry.py
@@ -57,9 +57,9 @@
 Subordinate data structures
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^
 
-.. autoclass:: TargetInfo()
+.. autoclass:: TargetInfo
 
-.. autoclass:: CenterToTargetList()
+.. autoclass:: CenterToTargetList
 
 Enums of special values
 ^^^^^^^^^^^^^^^^^^^^^^^
@@ -285,7 +285,6 @@ def make_container(
 
 # {{{ geometry data
 
-@dataclass_array_container
 @dataclass(frozen=True)
 class TargetInfo:
     """Describes the internal structure of the QBX FMM's list of :attr:`targets`.

diff --git a/pytential/unregularized.py b/pytential/unregularized.py
@@ -34,7 +34,7 @@
 from meshmode.dof_array import DOFArray
 
 from pytential.source import LayerPotentialSourceBase
-from pytential.array_context import PyOpenCLArrayContext, dataclass_array_container
+from pytential.array_context import PyOpenCLArrayContext
 
 import logging
 logger = logging.getLogger(__name__)
@@ -344,7 +344,6 @@ def build_traversal(self):
         return FMMTraversalBuilder(self.array_context)
 
 
-@dataclass_array_container
 @dataclass(frozen=True)
 class _TargetInfo:
     """
@@ -455,13 +454,11 @@ def target_info(self):
                         ),
                     )
 
-        info = _TargetInfo(
-                targets=targets,
+        return _TargetInfo(
+                targets=actx.freeze(targets),
                 target_discr_starts=target_discr_starts,
                 ntargets=ntargets)
 
-        return actx.freeze(info)
-
 # }}}