From 78fda6a0e41f83ccdf1c15dd957c3e080ff79a0c Mon Sep 17 00:00:00 2001
From: Nils Kohl <nils.kohl@fau.de>
Date: Tue, 28 May 2024 14:09:45 +0200
Subject: [PATCH] Typing issue II in generate_all_operators.py.

---
 generate_all_operators.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/generate_all_operators.py b/generate_all_operators.py
index 33580f3..2cae2eb 100644
--- a/generate_all_operators.py
+++ b/generate_all_operators.py
@@ -22,7 +22,7 @@ from functools import partial
 import logging
 import os
 import re
-from typing import Callable, List, Optional, Sequence, Set, Tuple
+from typing import Callable, List, Optional, Sequence, Set, Tuple, Union
 
 import numpy as np
 import sympy as sp
@@ -537,7 +537,7 @@ def all_operators(
     opts: List[Tuple[Set[Opts], LoopStrategy, str]],
     type_descriptor: HOGType,
     blending: GeometryMap,
-    geometries: Set[ElementGeometry],
+    geometries: Set[Union[TriangleElement, TetrahedronElement]],
 ) -> List[OperatorInfo]:
     P1 = LagrangianFunctionSpace(1, symbolizer)
     P1Vector = TensorialVectorFunctionSpace(P1)
-- 
GitLab