Merge pull request #184 from openego/features/depricated_python_operator

Update depricated PythonOperator import

CHANGELOG.rst

@@ -252,6 +252,9 @@ Added
   `PR #903 <https://github.com/openego/eGon-data/pull/903>`_
 * Add low flex scenario 'eGon2035_lowflex'
   `#822 <https://github.com/openego/eGon-data/issues/822>`_
+* Fix depricated python-operator import
+  `#184 <https://github.com/openego/powerd-data/pull/184>`_
+
 
 .. _PR #159: https://github.com/openego/eGon-data/pull/159
 .. _PR #703: https://github.com/openego/eGon-data/pull/703

src/egon/data/datasets/__init__.py

@@ -9,7 +9,7 @@
 import re
 
 from airflow.models.baseoperator import BaseOperator as Operator
-from airflow.operators.python_operator import PythonOperator
+from airflow.operators.python import PythonOperator
 from sqlalchemy import Column, ForeignKey, Integer, String, Table, orm, tuple_
 from sqlalchemy.ext.declarative import declarative_base
 

src/egon/data/datasets/electricity_demand_timeseries/hh_profiles.py

@@ -126,7 +126,7 @@
 import os
 import random
 
-from airflow.operators.python_operator import PythonOperator
+from airflow.operators.python import PythonOperator
 from sqlalchemy import ARRAY, Column, Float, Integer, String
 from sqlalchemy.dialects.postgresql import CHAR, INTEGER, REAL
 from sqlalchemy.ext.declarative import declarative_base

src/egon/data/datasets/emobility/motorized_individual_travel/__init__.py

@@ -77,7 +77,7 @@
 import os
 import tarfile
 
-from airflow.operators.python_operator import PythonOperator
+from airflow.operators.python import PythonOperator
 from psycopg2.extensions import AsIs, register_adapter
 import numpy as np
 import pandas as pd

src/egon/data/datasets/heat_supply/individual_heating.py

@@ -184,7 +184,7 @@
 import os
 import random
 
-from airflow.operators.python_operator import PythonOperator
+from airflow.operators.python import PythonOperator
 from psycopg2.extensions import AsIs, register_adapter
 from sqlalchemy import ARRAY, REAL, Column, Integer, String
 from sqlalchemy.ext.declarative import declarative_base