Correct Import Path for OrbiterMultiCronTimetable (#34)

* fix: update timetable w/ plugins reference * bump version * update doctest
astronomer · Nov 22, 2024 · 60ce620 · 60ce620
1 parent 098b6b2
commit 60ce620
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/orbiter/__init__.py b/orbiter/__init__.py
@@ -3,7 +3,7 @@
 import re
 from typing import Any, Tuple
 
-__version__ = "1.3.1"
+__version__ = "1.3.2"
 
 version = __version__
 

diff --git a/orbiter/objects/project.py b/orbiter/objects/project.py
@@ -218,8 +218,8 @@ def add_dags(self, dags: OrbiterDAG | Iterable[OrbiterDAG]) -> "OrbiterProject":
         OrbiterRequirement(names=[BashOperator], package=apache-airflow, module=airflow.operators.bash, sys_package=None),
         OrbiterRequirement(names=[send_smtp_notification], package=apache-airflow-providers-smtp, module=airflow.providers.smtp.notifications.smtp, sys_package=None),
         OrbiterRequirement(names=[TaskGroup], package=apache-airflow, module=airflow.utils.task_group, sys_package=None),
-        OrbiterRequirement(names=[MultiCronTimetable], package=croniter, module=multi_cron_timetable, sys_package=None),
-        OrbiterRequirement(names=[DateTime,Timezone], package=pendulum, module=pendulum, sys_package=None)],
+        OrbiterRequirement(names=[DateTime,Timezone], package=pendulum, module=pendulum, sys_package=None),
+        OrbiterRequirement(names=[MultiCronTimetable], package=croniter, module=plugins.multi_cron_timetable, sys_package=None)],
         pools=['foo'],
         connections=['SMTP', 'foo'],
         variables=['foo'],

diff --git a/orbiter/objects/timetables/multi_cron_timetable.py b/orbiter/objects/timetables/multi_cron_timetable.py
@@ -237,7 +237,7 @@ class OrbiterMultiCronTimetable(OrbiterTimetable):
     imports: ImportList = [
         OrbiterRequirement(
             package="croniter",
-            module="multi_cron_timetable",
+            module="plugins.multi_cron_timetable",
             names=["MultiCronTimetable"],
         )
     ]