PyCosmoMMF 0.1.2__py3-none-any.whl → 0.2.0__py3-none-any.whl

pycosmommf/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.1.2'
-__version_tuple__ = version_tuple = (0, 1, 2)
+__version__ = version = '0.2.0'
+__version_tuple__ = version_tuple = (0, 2, 0)

pycosmommf/filter.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import numba as nb
 import numpy as np
 
-jit_compiler = nb.njit(fastmath=True)
+jit_compiler = nb.njit(parallel=True, fastmath=True)
 
 
 def wavevectors3D(dims, box_size=(2 * np.pi, 2 * np.pi, 2 * np.pi)):
@@ -36,7 +36,7 @@ def wavevectors3D(dims, box_size=(2 * np.pi, 2 * np.pi, 2 * np.pi)):
 
 
 @jit_compiler
-def kspace_gaussian_filter(R_S, kv):
+def kspace_gaussian_filter(R_S, kv):  # pragma: no cover
     """
     create a Gaussian filter in k-space.
 
@@ -65,7 +65,7 @@ def kspace_gaussian_filter(R_S, kv):
 
 
 @jit_compiler
-def kspace_top_hat_filter(R_S, kv):
+def kspace_top_hat_filter(R_S, kv):  # pragma: no cover
     """
     create a top-hat filter in k-space.
 

pycosmommf/hessian.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import numba as nb
 import numpy as np
 
-jit_compiler = nb.njit(fastmath=True)
+jit_compiler = nb.njit(parallel=True, fastmath=True)
 
 
 def fast_hessian_from_smoothed(f_Rn, R_S, kv):
@@ -31,7 +31,7 @@ def fast_hessian_from_smoothed(f_Rn, R_S, kv):
     kx, ky, kz = kv[0], kv[1], kv[2]
 
     @jit_compiler
-    def perform_loop(hessian):
+    def perform_loop(hessian):  # pragma: no cover
         """
         This function is a loop that calculates the hessian matrix
         for each point in the 3D array.

pycosmommf/signatures.py

@@ -6,11 +6,11 @@ import numpy as np
 from .filter import smooth_gauss, smooth_loggauss, wavevectors3D
 from .hessian import fast_hessian_from_smoothed
 
-jit_compiler = nb.njit
+jit_compiler = nb.njit(parallel=True, fastmath=True)
 
 
 @jit_compiler
-def signatures_from_hessian(hessian):
+def signatures_from_hessian(hessian):  # pragma: no cover
     """
     Function to calculate the signatures from a given hessian.
 
@@ -115,7 +115,7 @@ def maximum_signature(Rs, field, alg="NEXUSPLUS", eps=1e-8):
     sigmax = np.ones((nx, ny, nz, 3), dtype=np.float32) * eps
 
     @jit_compiler
-    def perform_loop(sigmax, sigs_Rn, nx, ny, nz):
+    def perform_loop(sigmax, sigs_Rn, nx, ny, nz):  # pragma: no cover
         for ix in nb.prange(nx):
             for iy in range(ny):
                 for iz in range(nz):

pycosmommf/tagging.py

@@ -2,14 +2,11 @@ from __future__ import annotations
 
 import functools
 
-import numba as nb
 import numpy as np
 import skimage.measure
 
-jit_compiler = nb.njit(fastmath=True)
 
-
-def make_the_clusbool(delta, max_sigs, verbose, Δ):
+def make_the_clusbool(delta, max_sigs, Δ):  # pragma: no cover
     """
     make_the_clusbool() - Documentation
 
@@ -60,9 +57,7 @@ def make_the_clusbool(delta, max_sigs, verbose, Δ):
     S_th = signature_thresholds[np.abs(virialized_fractions - 0.5).argmin()]
     clusbool = max_sigs[:, :, :, 0] > S_th
 
-    if verbose:
-        return clusbool, S_th, signature_thresholds, virialized_fractions
-    return clusbool
+    return clusbool, S_th, signature_thresholds, virialized_fractions
 
 
 def calc_mass_change(sig_vec, delta_vec, Smin, Smax):
@@ -141,12 +136,13 @@ def calc_structure_bools(
     # - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
 
     if clusbool is None:
-        if verbose:
-            clusbool, cluster_thresh, S_clus, f_vir_clus = make_the_clusbool(
-                delta, max_sigs, verbose, Δ
-            )
-        else:
-            clusbool = make_the_clusbool(delta, max_sigs, verbose, Δ)
+        clusbool, cluster_thresh, S_clus, f_vir_clus = make_the_clusbool(
+            delta, max_sigs, Δ
+        )
+    else:
+        cluster_thresh = None
+        S_clus = None
+        f_vir_clus = None
 
     # - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
     #                 Step 2. Create Filament Boolean Filter

{pycosmommf-0.1.2.dist-info → pycosmommf-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pycosmommf
-Version: 0.1.2
+Version: 0.2.0
 Summary: A package for identifying structures in the cosmic web.
 Project-URL: Homepage, https://github.com/James11222/pycosmommf
 Project-URL: Bug Tracker, https://github.com/James11222/pycosmommf/issues

pycosmommf-0.2.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+pycosmommf/__init__.py,sha256=pmN6TzcwQ_VkHOjiSTmPvrNjA-uFQYQmTH1tiE4BNNU,356
+pycosmommf/_version.py,sha256=H-qsvrxCpdhaQzyddR-yajEqI71hPxLa4KxzpP3uS1g,411
+pycosmommf/_version.pyi,sha256=j5kbzfm6lOn8BzASXWjGIA1yT0OlHTWqlbyZ8Si_o0E,118
+pycosmommf/filter.py,sha256=UjkwIA22uhWPKTPq_r9bXrQzEv8_xBcwl9GQQcMffjg,4502
+pycosmommf/hessian.py,sha256=WiNPcfmxU3MG2D0w0bRf3iRL-2RJBWvgsXI-ZxBIrFU,1853
+pycosmommf/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pycosmommf/signatures.py,sha256=NJ36JboFgP-9oNW6Dc3-6PY61xSMATkehapEZVaxX2c,4039
+pycosmommf/tagging.py,sha256=c7POLpklFzobU-OQbIwmN6AM84yvbkYJ3VJ88gBz9xM,9768
+pycosmommf/utils.py,sha256=MYkAmf8PtifAM8R2oQPqwokdCUeuORkwO7tgiVWHEnY,1680
+pycosmommf-0.2.0.dist-info/METADATA,sha256=qbq5z4Zqk4H-vKhkk-gyjXwSx1ivntUbEmyj0OY_D1s,4426
+pycosmommf-0.2.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+pycosmommf-0.2.0.dist-info/licenses/LICENSE,sha256=IOy0BRQvhUnA4tFoZ1ZxLfQqx-6fkT6CNQdcma5WVQs,1053
+pycosmommf-0.2.0.dist-info/RECORD,,

pycosmommf-0.1.2.dist-info/RECORD

@@ -1,13 +0,0 @@
-pycosmommf/__init__.py,sha256=pmN6TzcwQ_VkHOjiSTmPvrNjA-uFQYQmTH1tiE4BNNU,356
-pycosmommf/_version.py,sha256=SFCDdrYA67D1Je-jHgVVh4LOopkPvuV6NMtqSJ7Tfhg,411
-pycosmommf/_version.pyi,sha256=j5kbzfm6lOn8BzASXWjGIA1yT0OlHTWqlbyZ8Si_o0E,118
-pycosmommf/filter.py,sha256=f9fH2LcmA-5Ysc42z8JLBs8gBKpNwGMYn5aVd6H8lEg,4447
-pycosmommf/hessian.py,sha256=-LVq4Zf2b-gzetPfx5vXn6qIwgLyqnCIOHOEhWStYU0,1818
-pycosmommf/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pycosmommf/signatures.py,sha256=cr1EmvFlDWxa0LUj3qk-mgih6mhxO0EZQQT01IIneA4,3969
-pycosmommf/tagging.py,sha256=bG9rnmlarbm8zmHV7mHUv2POiyQrm1F9eZ9OkyY0CME,9893
-pycosmommf/utils.py,sha256=MYkAmf8PtifAM8R2oQPqwokdCUeuORkwO7tgiVWHEnY,1680
-pycosmommf-0.1.2.dist-info/METADATA,sha256=Nx2ZXlP1bCnNM6xpSKWwJbN1EtmnVK0muAjdNVmrZ3U,4426
-pycosmommf-0.1.2.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-pycosmommf-0.1.2.dist-info/licenses/LICENSE,sha256=IOy0BRQvhUnA4tFoZ1ZxLfQqx-6fkT6CNQdcma5WVQs,1053
-pycosmommf-0.1.2.dist-info/RECORD,,