Merge pull request #8 from cody-scott/change-empty

cody-scott · Oct 2, 2024 · 846d2fc · 846d2fc
2 parents 1cf70f0 + 7b23ca8
commit 846d2fc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/dagster_mssql_bcp/bcp_core/bcp_io_manager_core.py b/src/dagster_mssql_bcp/bcp_core/bcp_io_manager_core.py
@@ -1,6 +1,6 @@
 from uuid import uuid4
 
-from dagster import ConfigurableIOManager, InputContext, OutputContext
+from dagster import ConfigurableIOManager, InputContext, OutputContext, get_dagster_logger
 
 from .asset_schema import AssetSchema
 from .mssql_connection import connect_mssql
@@ -35,7 +35,8 @@ def load_input(self, context: InputContext):
         raise NotImplementedError
 
     def handle_output(self, context: OutputContext, obj):
-        if self.check_empty(obj):
+        if obj is None:
+            get_dagster_logger().info("No data to load")
             return
 
         bcp_manager = self.get_bcp(