@@ -62,15 +62,15 @@ class BasePrep(KeyedDataAction):
62
62
def getInputSchema (self ) -> KeyedDataSchema :
63
63
yield from (
64
64
(column , Vector | Scalar | HealSparseMap | Tensor )
65
- for column in set (self .keysToLoad ).union (self .vectorKeys )
65
+ for column in sorted ( set (self .keysToLoad ).union (self .vectorKeys ) )
66
66
)
67
67
for action in self .selectors :
68
68
yield from action .getInputSchema ()
69
69
70
70
def getOutputSchema (self ) -> KeyedDataSchema :
71
71
return (
72
72
(column , Vector | Scalar | HealSparseMap | Tensor )
73
- for column in set (self .keysToLoad ).union (self .vectorKeys )
73
+ for column in sorted ( set (self .keysToLoad ).union (self .vectorKeys ) )
74
74
)
75
75
76
76
def __call__ (self , data : KeyedData , ** kwargs ) -> KeyedData :
@@ -82,7 +82,7 @@ def __call__(self, data: KeyedData, **kwargs) -> KeyedData:
82
82
else :
83
83
mask *= subMask # type: ignore
84
84
result : dict [str , Any ] = {}
85
- for key in set (self .keysToLoad ).union (self .vectorKeys ):
85
+ for key in sorted ( set (self .keysToLoad ).union (self .vectorKeys ) ):
86
86
formattedKey = key .format_map (kwargs )
87
87
result [formattedKey ] = cast (Vector , data [formattedKey ])
88
88
if mask is not None :
@@ -101,8 +101,8 @@ def addInputSchema(self, inputSchema: KeyedDataSchema) -> None:
101
101
existing .append (name )
102
102
if typ == Vector :
103
103
existingVectors .append (name )
104
- self .keysToLoad = set (existing )
105
- self .vectorKeys = set (existingVectors )
104
+ self .keysToLoad = sorted ( set (existing ) )
105
+ self .vectorKeys = sorted ( set (existingVectors ) )
106
106
107
107
108
108
class BaseProcess (KeyedDataAction ):
0 commit comments