supplement: restrict polars threads

wwood · Jun 2, 2024 · 759e5b0 · 759e5b0
1 parent ee31638
commit 759e5b0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/singlem/supplement.py b/singlem/supplement.py
@@ -827,6 +827,9 @@ def supplement(self, **kwargs):
         if len(kwargs) > 0:
             raise Exception("Unexpected arguments detected: %s" % kwargs)
 
+        os.environ["POLARS_MAX_THREADS"] = str(threads)
+        import polars as pl
+
         if new_genome_fasta_files is None and new_genome_fasta_files_list is None:
             raise Exception("Must provide either --new-genome-fasta-files or --new-genome-fasta-files-list")
         elif new_genome_fasta_files is not None and new_genome_fasta_files_list is not None: