Skip to content

Commit 79332eb

Browse files
committed
fix fit signature
1 parent cde6ab1 commit 79332eb

1 file changed

Lines changed: 11 additions & 7 deletions

File tree

src/pasteur/transform.py

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def fit(
6464
self,
6565
data: pd.Series | pd.DataFrame,
6666
ref: pd.Series | pd.DataFrame | None = None,
67-
) -> Attribute | None:
67+
):
6868
pass
6969

7070
def reduce(self, other: "RefTransformer"):
@@ -103,10 +103,10 @@ class SeqTransformer(Transformer):
103103
They can also push columns upstream to parents, through context tables.
104104
105105
Event-based data is sequential. The Sequential transformers may require the
106-
order of each row. For this case, the main Sequence Transformer, which is named
107-
the sequencer, is processed first and returns an additional data column and
106+
order of each row. For this case, the main Sequence Transformer, which is named
107+
the sequencer, is processed first and returns an additional data column and
108108
attribute during fitting. This column and attribute are fed to the other
109-
sequence transformers.
109+
sequence transformers.
110110
"""
111111

112112
def fit(
@@ -134,7 +134,9 @@ def fit_transform(
134134
ids: pd.DataFrame | None = None,
135135
seq_val: SeqValue | None = None,
136136
seq: pd.Series | None = None,
137-
) -> tuple[pd.DataFrame, dict[str, pd.DataFrame]] | tuple[pd.DataFrame, dict[str, pd.DataFrame], pd.Series]:
137+
) -> tuple[pd.DataFrame, dict[str, pd.DataFrame]] | tuple[
138+
pd.DataFrame, dict[str, pd.DataFrame], pd.Series
139+
]:
138140
self.fit(table, data, ref, ids, seq_val, seq)
139141
return self.transform(data, ref, ids, seq)
140142

@@ -144,15 +146,17 @@ def transform(
144146
ref: dict[str, pd.DataFrame] | None = None,
145147
ids: pd.DataFrame | None = None,
146148
seq: pd.Series | None = None,
147-
) -> tuple[pd.DataFrame, dict[str, pd.DataFrame]] | tuple[pd.DataFrame, dict[str, pd.DataFrame], pd.Series]:
149+
) -> tuple[pd.DataFrame, dict[str, pd.DataFrame]] | tuple[
150+
pd.DataFrame, dict[str, pd.DataFrame], pd.Series
151+
]:
148152
raise NotImplementedError()
149153

150154
def reverse(
151155
self,
152156
data: pd.DataFrame,
153157
ctx: dict[str, pd.DataFrame],
154158
ref: dict[str, pd.DataFrame] | None = None,
155-
ids: pd.DataFrame | None = None
159+
ids: pd.DataFrame | None = None,
156160
) -> pd.DataFrame:
157161
"""When reversing, the data column contains encoded data, whereas the ref
158162
column contains decoded/original data. Therefore, the referred columns have

0 commit comments

Comments
 (0)