diff --git a/experiments/airflow/dags/elasticity_pricing_dag.py b/experiments/airflow/dags/elasticity_pricing_dag.py index 4d6830d..f738537 100644 --- a/experiments/airflow/dags/elasticity_pricing_dag.py +++ b/experiments/airflow/dags/elasticity_pricing_dag.py @@ -22,7 +22,7 @@ from procesing.steps import ( ComputeDemandForChunksStep, AggregatePriceLogsStep, ComputeElasticityStep, - BuildStateSpaceStep, + # BuildStateSpaceStep, FitPricingFunctionStep, PredictPricesStep, ) @@ -38,7 +38,7 @@ default_args = { def get_provider(): """Factory to create composite provider""" - class CompositeProvider(SupabaseProvider, BackendAPIProvider): + class CompositeProvider(SupabaseProvider, BackendAPIProvider): # TODO: Fix this into one global provider singelton instead of multiple inheritance declarations acoss the codebase def __init__(self): SupabaseProvider.__init__(self) BackendAPIProvider.__init__(self) diff --git a/experiments/procesing/pipelines.py b/experiments/procesing/pipelines.py index 3b86654..c2e01e1 100644 --- a/experiments/procesing/pipelines.py +++ b/experiments/procesing/pipelines.py @@ -111,6 +111,7 @@ def full_pipeline(context: PipelineContext): return elasticity_df, prices_df + if __name__ == '__main__': class Provider(SupabaseProvider, BackendAPIProvider):