View and Set N Jobs (#1029)

* [Style] isort preference applied

* [Style] isort preference applied

* reset n_jobs to default after testing

* simplify imports

Author: DhruvSrikanth

tests/units/utilities/test_distribution.py

@@ -16,9 +16,23 @@
     LocalDaskDistributor,
     MultiprocessingDistributor,
 )
+from tsfresh.utilities.profiling import get_n_jobs, set_n_jobs
 
 
 class MultiprocessingDistributorTestCase(TestCase):
+    def test_n_jobs(self):
+        curr = get_n_jobs()
+        self.assertEqual(curr, get_n_jobs())
+
+        set_n_jobs(2)
+        self.assertEqual(2, get_n_jobs())
+
+        set_n_jobs(4)
+        self.assertEqual(4, get_n_jobs())
+
+        set_n_jobs(curr)
+        self.assertEqual(curr, get_n_jobs())
+
     def test_partition(self):
 
         distributor = MultiprocessingDistributor(n_workers=1)

tsfresh/utilities/profiling.py

@@ -10,6 +10,8 @@
 import logging
 import pstats
 
+from tsfresh import defaults
+
 _logger = logging.getLogger(__name__)
 
 
@@ -66,3 +68,26 @@ def end_profiling(profiler, filename, sorting=None):
             "[calculate_ts_features] Finished profiling of time series feature extraction"
         )
         f.write(s.getvalue())
+
+
+def get_n_jobs():
+    """
+    Get the number of jobs to use for parallel processing.
+
+    :return: The number of jobs to use for parallel processing.
+    :rtype: int
+    """
+    return defaults.N_PROCESSES
+
+
+def set_n_jobs(n_jobs):
+    """
+    Set the number of jobs to use for parallel processing.
+
+    :param n_jobs: The number of jobs to use for parallel processing.
+    :type n_jobs: int
+
+    :return: None
+    :rtype: None
+    """
+    defaults.N_PROCESSES = n_jobs