Organize code

memiiso · Sep 8, 2024 · 13a97d0 · 13a97d0
1 parent 79b0dda
commit 13a97d0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/opendbt/__init__.py b/opendbt/__init__.py
@@ -17,10 +17,10 @@
 # ================================================================================================================
 from opendbt.overrides import common
 
-if Version(get_dbt_version().to_version_string(skip_matcher=True)) > Version("1.8.0"):
-    dbt.task.docs.generate.GenerateTask = common.OpenDbtGenerateTask
-else:
+if Version(get_dbt_version().to_version_string(skip_matcher=True)) < Version("1.8.0"):
     dbt.task.generate.GenerateTask = common.OpenDbtGenerateTask
+else:
+    dbt.task.docs.generate.GenerateTask = common.OpenDbtGenerateTask
 factory.FACTORY = common.OpenDbtAdapterContainer()
 
 

diff --git a/opendbt/overrides/common.py b/opendbt/overrides/common.py
@@ -9,12 +9,12 @@
 
 from dbt.version import get_installed_version as get_dbt_version
 
-if Version(get_dbt_version().to_version_string(skip_matcher=True)) > Version("1.8.0"):
-    from dbt.task.docs.generate import GenerateTask
-    from opendbt.overrides.dbt18 import AdapterContainerDbtOverride
-else:
+if Version(get_dbt_version().to_version_string(skip_matcher=True)) < Version("1.8.0"):
     from dbt.task.generate import GenerateTask
     from opendbt.overrides.dbt17 import AdapterContainerDbtOverride
+else:
+    from dbt.task.docs.generate import GenerateTask
+    from opendbt.overrides.dbt18 import AdapterContainerDbtOverride
 
 
 class OpenDbtAdapterContainer(AdapterContainerDbtOverride):