mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-15 03:48:09 +00:00
chore(Linter): reformatted code with black (#211)
* chore(Linter): reformatted code with black * Create black.yaml
This commit is contained in:
+5
-4
@@ -4,8 +4,8 @@ import pandas as pd
|
||||
import pandera as pa
|
||||
from pandera.typing import DataFrame, Series
|
||||
|
||||
class EventFilter(ABC):
|
||||
|
||||
class EventFilter(ABC):
|
||||
@abstractmethod
|
||||
def get_event_start_times(self, returns: ReturnSeries) -> pd.DatetimeIndex:
|
||||
raise NotImplementedError
|
||||
@@ -17,12 +17,13 @@ class EventSchema(pa.SchemaModel):
|
||||
label: Series[int]
|
||||
returns: Series[float]
|
||||
|
||||
|
||||
EventsDataFrame = DataFrame[EventSchema]
|
||||
|
||||
|
||||
class EventLabeller(ABC):
|
||||
|
||||
@abstractmethod
|
||||
def label_events(self, event_start_times: pd.DatetimeIndex, returns: ReturnSeries) -> tuple[EventsDataFrame, ForwardReturnSeries]:
|
||||
def label_events(
|
||||
self, event_start_times: pd.DatetimeIndex, returns: ReturnSeries
|
||||
) -> tuple[EventsDataFrame, ForwardReturnSeries]:
|
||||
raise NotImplementedError
|
||||
|
||||
|
||||
Reference in New Issue
Block a user