make pyranges optional again (#77)

kipoi · Jun 4, 2020 · cf4c42e · cf4c42e
1 parent 4635665
commit cf4c42e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kipoiseq/extractors/protein.py b/kipoiseq/extractors/protein.py
@@ -7,7 +7,6 @@
 from kipoiseq.extractors.vcf_seq import VariantSeqExtractor
 from kipoiseq.extractors.vcf_matching import SingleVariantMatcher
 from typing import List
-import pyranges
 import pandas as pd
 
 # TODO: convert print to logs
@@ -74,6 +73,7 @@ def _read_cds(gtf_file, duplicate_attr=False):
         Read, extract and filter valid cds from the given gtf_file
         :param gtf_file:
         """
+        import pyranges
         df = pyranges.read_gtf(gtf_file, as_df=True,
                                duplicate_attr=duplicate_attr)
         cds = CDSFetcher._get_cds_from_gtf(df)
@@ -229,6 +229,7 @@ def __init__(self, gtf_file, fasta_file, vcf_file):
         self.vcf_file = str(vcf_file)
         self.cds_fetcher = CDSFetcher(self.gtf_file)
         # dataframe to pyranges
+        import pyranges
         pr_cds = pyranges.PyRanges(self.cds_fetcher.cds.reset_index())
         # match variant with transcript_id
         self.single_variant_matcher = SingleVariantMatcher(