chore: clean up definition of composite class of providers

2026-05-31 08:33:36 +00:00 · 2025-12-11 21:48:05 +01:00
parent 9ca6468924
commit c80101aa6e
1 changed files with 5 additions and 4 deletions
--- a/experiments/airflow/dags/surge_pricing_factory.py
+++ b/experiments/airflow/dags/surge_pricing_factory.py
@@ -29,11 +29,12 @@ DEFAULT_ARGS = {
    'retry_delay': timedelta(minutes=5),
 }

-def _get_provider():
-    class CompositeProvider(SupabaseProvider, BackendAPIProvider):
+class CompositeProvider(SupabaseProvider, BackendAPIProvider):
    def __init__(self):
        SupabaseProvider.__init__(self)
        BackendAPIProvider.__init__(self)
+
+def _get_provider():
    return CompositeProvider()

 def _make_task_callables(store_mode: str):