code-loader 0.2.84.dev30__py3-none-any.whl → 0.2.84.dev31__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -13,9 +13,7 @@ class LeapLoaderParallelizedBase(ABC):
13
13
  def __init__(self, code_path: str, code_entry_name: str,
14
14
  n_workers: Optional[int] = 2, max_ready_results_in_queue: int = 128,
15
15
  multiprocessing_context: Optional[str] = None) -> None:
16
- self.multiprocessing_context = multiprocessing
17
- if multiprocessing_context is not None:
18
- self.multiprocessing_context = multiprocessing.get_context(multiprocessing_context)
16
+ self.multiprocessing_context = multiprocessing.get_context(multiprocessing_context)
19
17
 
20
18
  self.code_entry_name = code_entry_name
21
19
  self.code_path = code_path
@@ -37,7 +35,7 @@ class LeapLoaderParallelizedBase(ABC):
37
35
  memory_usage_in_bytes = p.memory_info().rss
38
36
  total_memory_in_bytes = psutil.virtual_memory().total
39
37
 
40
- n_workers = min(int(multiprocessing.cpu_count()),
38
+ n_workers = min(int(self.multiprocessing_context.cpu_count()),
41
39
  int(total_memory_in_bytes * 0.7 / memory_usage_in_bytes))
42
40
  n_workers = max(n_workers, 1)
43
41
  return n_workers
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: code-loader
3
- Version: 0.2.84.dev30
3
+ Version: 0.2.84.dev31
4
4
  Summary:
5
5
  Home-page: https://github.com/tensorleap/code-loader
6
6
  License: MIT
@@ -20,7 +20,7 @@ code_loader/helpers/instancesegmentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5
20
20
  code_loader/helpers/instancesegmentation/utils.py,sha256=bl4nlg7mV8CZyPl8grGAE5603TFVM7PTZyM3BRMbXBg,930
21
21
  code_loader/leap_binder/__init__.py,sha256=Oe7DNj6RmgX-tMruOrI7GYvf3SyAnzOxugF2e117_Z8,93
22
22
  code_loader/leap_binder/leapbinder.py,sha256=WjC-UA5xeY9UZtMqJQOxjidlFcWURC9mmy7OrQy1IY0,6692
23
- code_loader/leap_loader_parallelized_base.py,sha256=_HNbjRK42wRM1V252ETVzibHW8bofFK8-Kq-It5fRks,4758
23
+ code_loader/leap_loader_parallelized_base.py,sha256=BsZW7nvWZY9Rgu5llyOgoaZ1Q9QNhPNiyyOkJtnknFY,4664
24
24
  code_loader/leaploader.py,sha256=MnQeNOwNgkQItCWFiWT6xBxAsqjNbcTOPk8HIP8v19o,15370
25
25
  code_loader/metric_calculator_parallelized.py,sha256=z1PmIRfizTkWWAIZr4knFYGEqZvlJWEyVluSSMiRxjg,2164
26
26
  code_loader/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -29,7 +29,7 @@ code_loader/samples_generator_parallelized.py,sha256=p0hQ6tKL3GI9nlDKa1GDFD1Gf3S
29
29
  code_loader/utils.py,sha256=WUcM97OuxrhfLCRPoH9EbXrxajNpYgX1CTMc3_PXtYU,1736
30
30
  code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
31
31
  code_loader/visualizers/default_visualizers.py,sha256=HqWx2qfTrroGl2n8Fpmr_4X-rk7tE2oGapjO3gzz4WY,2226
32
- code_loader-0.2.84.dev30.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
33
- code_loader-0.2.84.dev30.dist-info/METADATA,sha256=BCPqPb9uQlhaAGP8U9ZOQFoZMFtAtMYaE4h6bnYge2c,953
34
- code_loader-0.2.84.dev30.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
35
- code_loader-0.2.84.dev30.dist-info/RECORD,,
32
+ code_loader-0.2.84.dev31.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
33
+ code_loader-0.2.84.dev31.dist-info/METADATA,sha256=4Yg4zR5adl6atAM6iH43eBU2_lJxWmNyVgLoyQ6ygkk,953
34
+ code_loader-0.2.84.dev31.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
35
+ code_loader-0.2.84.dev31.dist-info/RECORD,,