remove negatives

mpkrass7 · Jun 20, 2024 · 4a0135c · 4a0135c
1 parent b6df269
commit 4a0135c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...utter.repo_name }}/src/{{ cookiecutter.python_package }}/pipelines/preprocessing/nodes.py b/...utter.repo_name }}/src/{{ cookiecutter.python_package }}/pipelines/preprocessing/nodes.py
@@ -78,6 +78,8 @@ def create_or_update_modeling_dataset(modeling_dataset_name: str, #TODO: change
 
         staging_data.fillna(0, inplace=True)
 
+        staging_data = staging_data["viewDiff"].apply(lambda x: max(x, 0))
+
         dataset = dr.Dataset.create_version_from_in_memory_data(modeling_dataset_id, staging_data)
 
     return str(dataset.id)