Operators: refactoring

cern-sis · Aug 2, 2024 · a1d0250 · a1d0250
1 parent 0bf939c
commit a1d0250
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dags/common/operators/sqlalchemy_operator.py b/dags/common/operators/sqlalchemy_operator.py
@@ -14,7 +14,7 @@ def get_session(conn_id: str):
 def sqlalchemy_task(conn_id: str):
     def decorator(func):
         @task(executor_config=kubernetes_executor_config)
-        def wrapper(*args, **kwargs):
+        def populate_database(*args, **kwargs):
             session = get_session(conn_id)
             try:
                 result = func(*args, session=session, **kwargs)
@@ -26,6 +26,6 @@ def wrapper(*args, **kwargs):
             finally:
                 session.close()
 
-        return wrapper
+        return populate_database
 
     return decorator