🎨 Modified imports

dyvenia · Nov 5, 2024 · 4e375d2 · 4e375d2
1 parent ebb6936
commit 4e375d2
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/tests/unit/orchestration/prefect/test_utils.py b/tests/unit/orchestration/prefect/test_utils.py
@@ -1,16 +1,9 @@
 import pendulum
 import pytest
 
-from viadot.utils import skip_test_on_missing_extra
+from viadot.orchestration.prefect.utils import DynamicDateHandler
 
 
-try:
-    from viadot.orchestration.prefect.utils import DynamicDateHandler
-except ImportError:
-    skip_test_on_missing_extra(
-        source_name="DynamicDateHandler", extra="dynamicDateHandler"
-    )
-
 ddh1 = DynamicDateHandler(
     ["<<", ">>"], dynamic_date_format="%Y%m%d", dynamic_date_timezone="Europe/Warsaw"
 )