Test sample_script calling on new Operator

orchestrate/dags/sample_script.py

@@ -0,0 +1,27 @@
+from airflow.decorators import dag
+from kubernetes.client import models as k8s
+from operators.datacoves.bash import DatacovesBashOperator
+
+default_args = {
+    "owner": "airflow",
+    "email": "some_user@exanple.com",
+    "email_on_failure": True,
+    "description": "Sample python dag",
+}
+
+
+@dag(
+    default_args={"start_date": "2021-01"},
+    description="sample_script",
+    schedule_interval="0 0 1 */12 *",
+    tags=["version_2"],
+    catchup=False,
+)
+def sample_script():
+    sample_script = DatacovesBashOperator(
+        task_id="sample_script",
+        bash_command=f"python orchestrate/python_scripts/sample_script.py",
+    )
+
+
+dag = sample_script()

orchestrate/python_scripts/sample_script.py

@@ -1,16 +1,3 @@
-import pandas as pd
+from orchestrate.some_module.main import MainRunner
 
-def print_sample_dataframe():
-
-    # Creating a simple DataFrame
-    data = {'Name': ['Alice', 'Bob', 'Charlie'],
-            'Age': [25, 30, 35],
-            'City': ['New York', 'San Francisco', 'Los Angeles']}
-
-    df = pd.DataFrame(data)
-
-    # Displaying the DataFrame
-    print("DataFrame created using Pandas:")
-    print(df)
-
-print_sample_dataframe()
+runner = MainRunner()

orchestrate/some_module/main.py

@@ -0,0 +1,3 @@
+class MainRunner:
+    def __init__(self) -> None:
+        print("SUCCESS")