try to resolve failing test

ERROR trtools/testsupport/sample_vcfs/associaTR/generate_traits.py - FileNotFoundError: [Errno 2] No such file or directory: 'samples.txt'
gymrek-lab · Oct 30, 2023 · aaddb38 · aaddb38
1 parent 38565e8
commit aaddb38
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/trtools/testsupport/sample_vcfs/associaTR/generate_traits.py b/trtools/testsupport/sample_vcfs/associaTR/generate_traits.py
@@ -1,11 +1,14 @@
 #!/usr/bin/env python3
 
+import pathlib
 import numpy as np
 import numpy.random
 
 seed = 2
 
-with open('samples.txt') as samples_file:
+SCRIPT_DIR = pathlib.Path(__file__).parent.resolve()
+
+with open(SCRIPT_DIR / 'samples.txt') as samples_file:
     samples = np.array([int(sample.strip()) for sample in samples_file.readlines() if 'IID' not in sample])
 n_samples = len(samples)