From 5360c8236376b542bc1fbd929e569e5171f9dacc Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Wed, 22 Nov 2023 15:14:48 +0900
Subject: [PATCH] fixed types
---
src/pystencilssfg/source_concepts/cpp/std_mdspan.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/src/pystencilssfg/source_concepts/cpp/std_mdspan.py b/src/pystencilssfg/source_concepts/cpp/std_mdspan.py
index d600387..70eeb8a 100644
--- a/src/pystencilssfg/source_concepts/cpp/std_mdspan.py
+++ b/src/pystencilssfg/source_concepts/cpp/std_mdspan.py
@@ -1,4 +1,4 @@
-from typing import Union
+from typing import Union, cast
import numpy as np
@@ -17,7 +17,7 @@ class StdMdspan(SrcField):
def __init__(self, identifer: str,
T: PsType,
- extents: tuple[int, str],
+ extents: tuple[int | str, ...],
extents_type: PsType = int,
reference: bool = False):
cpp_typestr = cpp_typename(T)
@@ -89,16 +89,16 @@ def mdspan_ref(field: Field, extents_type: type = np.uint32):
if field.layout != layout_string_to_tuple("soa", field.spatial_dimensions):
raise NotImplementedError("mdspan mapping is currently only available for structure-of-arrays fields")
- extents = []
+ extents : list[str | int] = []
for s in field.spatial_shape:
- extents += StdMdspan.dynamic_extent if isinstance(s, FieldShapeSymbol) else s
+ extents.append(StdMdspan.dynamic_extent if isinstance(s, FieldShapeSymbol) else cast(int, s))
if field.index_shape != (1,):
for s in field.index_shape:
extents += StdMdspan.dynamic_extent if isinstance(s, FieldShapeSymbol) else s
return StdMdspan(field.name, field.dtype,
- (StdMdspan.dynamic_extent, StdMdspan.dynamic_extent),
+ tuple(extents),
extents_type=extents_type,
reference=True)
--
GitLab