🐛 removed if statement

dyvenia · Oct 25, 2023 · d144bd6 · d144bd6
1 parent bd492c6
commit d144bd6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/tests/integration/flows/test_customer_gauge_to_adls.py b/tests/integration/flows/test_customer_gauge_to_adls.py
@@ -88,4 +88,7 @@ def test_customer_gauge_to_adls_run_flow_validation_failure(mocked_class):
         flow.run()
     except ValidationError:
         pass
+
+    os.remove("test_customer_gauge_to_adls_flow_run.parquet")
+    os.remove("test_customer_gauge_to_adls_flow_run.json")
 
diff --git a/viadot/flows/customer_gauge_to_adls.py b/viadot/flows/customer_gauge_to_adls.py
@@ -191,12 +191,6 @@ def gen_flow(self) -> Flow:
                 days=self.days,
                 flow=self,
             )
-
-        if self.validate_df_dict:
-            validation_task = validate_df.bind(
-                anonymized_df, tests=self.validate_df_dict, flow=self
-            )
-            validation_task.set_upstream(anonymized_df, flow=self)
 
             df_with_metadata = add_ingestion_metadata_task.bind(
                 anonymized_df, flow=self