removing unneeded package

HenryDayHall · Sep 29, 2023 · aa1fdfc · aa1fdfc
1 parent 4d5f4fa
commit aa1fdfc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_ParallelFormJets.py b/test/test_ParallelFormJets.py
@@ -1,5 +1,5 @@
 """ Module to test the code in ParallelFormJets"""
-import psutil
+import multiprocessing
 import pytest
 import numpy.testing as tst
 import shutil
@@ -217,7 +217,7 @@ def test_generate_pool():
             end_time = time.time() + 100
             jet_class = FormJets.Spectral
             jet_params = {"Bark": 3}
-            n_cores = psutil.cpu_count()
+            n_cores = multiprocessing.cpu_count()
             function = lambda x: x
             found = ParallelFormJets.generate_pool(eventWise_path, function,
                 function_args=[jet_class, "Bali", jet_params],