Fix tab

coderxio · Jul 31, 2024 · 1fa0a24 · 1fa0a24
1 parent 331d537
commit 1fa0a24
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/airflow/dags/sagerx.py b/airflow/dags/sagerx.py
@@ -178,8 +178,8 @@ def load_df_to_pg(df,schema_name:str,table_name:str,if_exists:str,dtype_name:str
     )
 
     if create_index and index_columns:
-            columns_str = ', '.join(index_columns)
-            engine.execute(f'CREATE INDEX IF NOT EXISTS idx_{table_name}_{"_".join(index_columns)} ON {schema_name}.{table_name} ({columns_str})')
+        columns_str = ', '.join(index_columns)
+        engine.execute(f'CREATE INDEX IF NOT EXISTS idx_{table_name}_{"_".join(index_columns)} ON {schema_name}.{table_name} ({columns_str})')
 
 def run_query_to_df(query:str) -> pd.DataFrame:
     from airflow.hooks.postgres_hook import PostgresHook