1
1
from __future__ import annotations
2
2
3
3
from collections .abc import Sequence
4
- from typing import Any , Literal , TypedDict
4
+ from typing import Literal , TypedDict , Union
5
5
6
6
__all__ = [
7
7
"BooleanAxis" ,
17
17
"WeightedStorage" ,
18
18
]
19
19
20
+ SupportedMetadata = Union [float , str , bool ]
21
+
20
22
21
23
def __dir__ () -> list [str ]:
22
24
return __all__
@@ -33,7 +35,7 @@ class _RequiredRegularAxis(TypedDict):
33
35
34
36
35
37
class RegularAxis (_RequiredRegularAxis , total = False ):
36
- metadata : dict [str , Any ]
38
+ metadata : dict [str , SupportedMetadata ]
37
39
38
40
39
41
class _RequiredVariableAxis (TypedDict ):
@@ -45,7 +47,7 @@ class _RequiredVariableAxis(TypedDict):
45
47
46
48
47
49
class VariableAxis (_RequiredVariableAxis , total = False ):
48
- metadata : dict [str , Any ]
50
+ metadata : dict [str , SupportedMetadata ]
49
51
50
52
51
53
class _RequiredCategoryStrAxis (TypedDict ):
@@ -55,7 +57,7 @@ class _RequiredCategoryStrAxis(TypedDict):
55
57
56
58
57
59
class CategoryStrAxis (_RequiredCategoryStrAxis , total = False ):
58
- metadata : dict [str , Any ]
60
+ metadata : dict [str , SupportedMetadata ]
59
61
60
62
61
63
class _RequiredCategoryIntAxis (TypedDict ):
@@ -65,15 +67,15 @@ class _RequiredCategoryIntAxis(TypedDict):
65
67
66
68
67
69
class CategoryIntAxis (_RequiredCategoryIntAxis , total = False ):
68
- metadata : dict [str , Any ]
70
+ metadata : dict [str , SupportedMetadata ]
69
71
70
72
71
73
class _RequiredBooleanAxis (TypedDict ):
72
74
type : Literal ["boolean" ]
73
75
74
76
75
77
class BooleanAxis (_RequiredBooleanAxis , total = False ):
76
- metadata : dict [str , Any ]
78
+ metadata : dict [str , SupportedMetadata ]
77
79
78
80
79
81
class IntStorage (TypedDict ):
@@ -117,4 +119,4 @@ class _RequiredHistogram(TypedDict):
117
119
118
120
119
121
class Histogram (_RequiredHistogram , total = False ):
120
- metadata : dict [str , Any ]
122
+ metadata : dict [str , SupportedMetadata ]
0 commit comments