Set test line to train line if null

curt-tigges · Oct 8, 2023 · 5b88da8 · 5b88da8
1 parent 6a42a89
commit 5b88da8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/classification.py b/utils/classification.py
@@ -299,7 +299,7 @@ def train_classifying_direction(
                 **kwargs,
             )
             if test_data is None:
-                test_data = train_data
+                test_line = train_line
             else:
                 test_line, _, _, _ = fitting_method(
                     test_data, test_pos, test_layer,