init

pipeline/run.py

@@ -0,0 +1,22 @@
+from pydantic import BaseModel, Field
+import sqlite3
+import pandas as pd
+from typing import Optional
+from .metadata import Metadata
+
+class Run(BaseModel):
+    # === FETCHERS ===
+    onet_conn: Optional[sqlite3.Connection] = None
+    onet_version: Optional[str] = None
+
+    oesm_df: Optional[pd.DataFrame] = None
+    oesm_version: Optional[str] = None
+
+    epoch_df: Optional[pd.DataFrame] = None
+    epoch_version: Optional[str] = None
+
+    # === ENRICHMENTS ===
+    task_estimateability_df: Optional[pd.DataFrame] = None
+    task_estimates_df: Optional[pd.DataFrame] = None
+
+    meta: Metadata = Field(default_factory=Metadata)