Skip to content

Commit b084b9c

Browse files
Move data_check to ABC datacollector class
1 parent f261386 commit b084b9c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

causal_testing/data_collection/data_collector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ class DataCollector(ABC):
1818

1919
def __init__(self, scenario: Scenario):
2020
self.scenario = scenario
21+
self.data_checked = False
2122

2223
@abstractmethod
2324
def collect_data(self, **kwargs) -> pd.DataFrame:
@@ -140,7 +141,6 @@ class ObservationalDataCollector(DataCollector):
140141
def __init__(self, scenario: Scenario, data: pd.DataFrame):
141142
super().__init__(scenario)
142143
self.data = data
143-
self.data_checked = False
144144

145145
def collect_data(self, **kwargs) -> pd.DataFrame:
146146
"""Read a pandas dataframe and filter to remove

0 commit comments

Comments
 (0)