fakecbed 0.3.3__py3-none-any.whl → 0.3.5__py3-none-any.whl

fakecbed/discretized.py

@@ -1482,6 +1482,11 @@ class CBEDPattern(fancytypes.PreSerializableAndUpdatable):
         illumination_support = self._illumination_support.cpu().detach().clone()
         illumination_support = illumination_support.numpy(force=True)
 
+        if self._disk_supports is None:
+            method_name = ("_calc_disk_supports"
+                           "_and_cache_select_intermediates")
+            method_alias = getattr(self, method_name)
+            self._disk_supports = method_alias(u_x, u_y)
         disk_supports = self._disk_supports.numpy(force=True)
 
         num_disks = self._num_disks
@@ -1701,11 +1706,13 @@ class CBEDPattern(fancytypes.PreSerializableAndUpdatable):
 
         apply_shot_noise = self._apply_shot_noise
         if apply_shot_noise == True:
-            np_rng = np.random.default_rng(self._rng_seed)
-            torch_rng_seed = np_rng.integers(low=0, high=2**32-1).item()
-            torch_rng = torch.Generator(device=u_x.device)
-            torch_rng = torch_rng.manual_seed(torch_rng_seed)
-            image = torch.poisson(image, torch_rng)
+            # ``torch.poisson`` was occasionally causing CUDA errors.
+            rng = np.random.default_rng(self._rng_seed)
+            image_dtype = image.dtype
+            image = image.numpy(force=True)
+            image = rng.poisson(image)
+            image = torch.from_numpy(image)
+            image = image.to(device=self._device, dtype=image_dtype)
 
         image = self._apply_detector_partition_inpainting(input_image=image)
             
@@ -1715,7 +1722,7 @@ class CBEDPattern(fancytypes.PreSerializableAndUpdatable):
             self._illumination_support = method_alias(u_x, u_y)
         illumination_support = self._illumination_support
 
-        image = image*illumination_support        
+        image = image*illumination_support
 
         coords_of_cold_pixels = self._cold_pixels
         L, R, B, T = self._mask_frame

fakecbed/version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.3'
-__version_tuple__ = version_tuple = (0, 3, 3)
+__version__ = version = '0.3.5'
+__version_tuple__ = version_tuple = (0, 3, 5)

{fakecbed-0.3.3.dist-info → fakecbed-0.3.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fakecbed
-Version: 0.3.3
+Version: 0.3.5
 Summary: A Python library for generating quickly images that imitate convergent beam electron diffraction patterns.
 Author-email: Matthew Fitzpatrick <matthew.rc.fitzpatrick@gmail.com>
 Project-URL: Homepage, https://mrfitzpa.github.io/fakecbed

fakecbed-0.3.5.dist-info/RECORD

@@ -0,0 +1,10 @@
+fakecbed/__init__.py,sha256=iP73WNV_HZgxiVUVdesbJqlrhDGESooh1hlBs8kox9M,1342
+fakecbed/discretized.py,sha256=qqI_F4Ax9oH6nAdXEOTTKji9CH52oEIrURAopSGauu8,99688
+fakecbed/shapes.py,sha256=TiqJBZI4JZib0Q6zfO0e-dh7ACjYhvAYUvEtIedoo74,116471
+fakecbed/tds.py,sha256=mk5359ErIhdpCoTdHq9r1-foKb8t3qCOHqNoabx3bQw,8066
+fakecbed/version.py,sha256=qmTn4w-xTKk8dVv-032b430C-ilxMTkr9Q0HieU18N8,511
+fakecbed-0.3.5.dist-info/licenses/LICENSE,sha256=N0P3pKtRMvfb64jmgJdlerg3TnS0a2QC776AzPsbZIg,35128
+fakecbed-0.3.5.dist-info/METADATA,sha256=PkGdAdfTboRbWK_BEvcMt4DoiUAxIN_Nd1IDe_--6dU,3023
+fakecbed-0.3.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fakecbed-0.3.5.dist-info/top_level.txt,sha256=l0jL8P0CjRiCSznYLnq7riaAnOmYaV3zjjjIJx9vjSQ,9
+fakecbed-0.3.5.dist-info/RECORD,,

fakecbed-0.3.3.dist-info/RECORD

@@ -1,10 +0,0 @@
-fakecbed/__init__.py,sha256=iP73WNV_HZgxiVUVdesbJqlrhDGESooh1hlBs8kox9M,1342
-fakecbed/discretized.py,sha256=zniskVmHRyxnR-P_O7qvyHuk2nnujUx1zDI99QDN53U,99379
-fakecbed/shapes.py,sha256=TiqJBZI4JZib0Q6zfO0e-dh7ACjYhvAYUvEtIedoo74,116471
-fakecbed/tds.py,sha256=mk5359ErIhdpCoTdHq9r1-foKb8t3qCOHqNoabx3bQw,8066
-fakecbed/version.py,sha256=cRYgYV4ttw-FMlrA4-5pzcSpTjS7X8uVa-nRTEADKW4,511
-fakecbed-0.3.3.dist-info/licenses/LICENSE,sha256=N0P3pKtRMvfb64jmgJdlerg3TnS0a2QC776AzPsbZIg,35128
-fakecbed-0.3.3.dist-info/METADATA,sha256=ntsj55iN8Ddoe9FtuB5FsY_tmvHzjB40iCoRtyrm35U,3023
-fakecbed-0.3.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fakecbed-0.3.3.dist-info/top_level.txt,sha256=l0jL8P0CjRiCSznYLnq7riaAnOmYaV3zjjjIJx9vjSQ,9
-fakecbed-0.3.3.dist-info/RECORD,,