Type SKELETONIZE_TEST_CASES

inducer · Aug 14, 2023 · 096a345 · 096a345
1 parent fe6c9f8
commit 096a345
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/test_linalg_skeletonization.py b/test/test_linalg_skeletonization.py
@@ -22,6 +22,7 @@
 
 from dataclasses import replace
 from functools import partial
+from typing import Sequence, Union
 import pytest
 
 import numpy as np
@@ -47,7 +48,7 @@
     PytestPyOpenCLArrayContextFactory,
     ])
 
-SKELETONIZE_TEST_CASES = [
+SKELETONIZE_TEST_CASES: Sequence[Union[extra.CurveTestCase, extra.TorusTestCase]] = [
         extra.CurveTestCase(
             name="ellipse",
             op_type="scalar",