Update setup

setup.py

@@ -16,7 +16,7 @@
     long_description=README,
     long_description_content_type="text/markdown",
     url="https://github.com/xin-huang/sstar",
-    author="Xin Huang, Joesf Hackl",
+    author="Xin Huang, Josef Hackl",
     author_email="[email protected]",
     license="Apache-2.0",
     classifiers=[
@@ -33,6 +33,7 @@
         "pandas",
         "rpy2",
         "scikit-allel",
+        "scikit-learn",
         "scipy",
         "statsmodels",
     ],

sstar/preprocess.py

@@ -238,6 +238,5 @@ def _output(res, tgt_samples, header, features, output_dir, output_prefix, outpu
                     f.write(f'{chrom}\t{start}\t{end}\t{sample}\t{out}\n')
 
 
-
 if __name__ == '__main__':
     process_data(vcf_file="examples/data/real_data/sstar.example.biallelic.snps.vcf.gz", ref_ind_file="examples/data/ind_list/ref.ind.list", tgt_ind_file="examples/data/ind_list/tgt.ind.list", anc_allele_file=None, feature_file="examples/pre-trained/test.features.yaml", output_dir="sstar/test", output_prefix="test", win_len=50000, win_step=10000, thread=1)