output shape change when windows=wholegnome

tskit-dev · Oct 11, 2024 · 45ac61e · 45ac61e
1 parent 6e415e5
commit 45ac61e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tskit/trees.py b/python/tskit/trees.py
@@ -8731,7 +8731,7 @@ def _genetic_relatedness_vector(
                         rng=random_state
                     )
 
-        if drop_windows or len(windows) == 2:
+        if drop_windows:
             U, D = U[0], D[0]
 
         return U, D