WIP

trane-dev · Dec 21, 2023 · 996be5c · 996be5c
1 parent 9462c1f
commit 996be5c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tests/test_target_values.py b/tests/test_target_values.py
@@ -9,7 +9,7 @@
 from trane.ops.transformation_ops import IdentityOp
 
 
-@pytest.fixture
+@pytest.fixture(scope="module")
 def data():
     num_rows = 100
     data = {

diff --git a/trane/core/utils.py b/trane/core/utils.py
@@ -47,6 +47,8 @@ def calculate_target_values(
             window_size=window_size,
             gap=window_size,
         ):
+            if dataslice.empty:
+                continue
             record = labeling_function(dataslice)
             records.append(
                 {