fakecbed 0.3.2__py3-none-any.whl → 0.3.4__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,12 @@ 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 = 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)
             

fakecbed/version.py

@@ -1,8 +1,13 @@
-# file generated by setuptools_scm
+# file generated by setuptools-scm
 # don't change, don't track in version control
+
+__all__ = ["__version__", "__version_tuple__", "version", "version_tuple"]
+
 TYPE_CHECKING = False
 if TYPE_CHECKING:
-    from typing import Tuple, Union
+    from typing import Tuple
+    from typing import Union
+
     VERSION_TUPLE = Tuple[Union[int, str], ...]
 else:
     VERSION_TUPLE = object
@@ -12,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.2'
-__version_tuple__ = version_tuple = (0, 3, 2)
+__version__ = version = '0.3.4'
+__version_tuple__ = version_tuple = (0, 3, 4)

{fakecbed-0.3.2.dist-info → fakecbed-0.3.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: fakecbed
-Version: 0.3.2
+Version: 0.3.4
 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
@@ -22,7 +22,7 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: distoptica>=0.3.3
-Requires-Dist: hyperspy[all]
+Requires-Dist: hyperspy
 Provides-Extra: tests
 Requires-Dist: pytest-cov; extra == "tests"
 Provides-Extra: examples
@@ -36,6 +36,7 @@ Requires-Dist: numpydoc; extra == "docs"
 Requires-Dist: docutils; extra == "docs"
 Provides-Extra: all
 Requires-Dist: fakecbed[docs,examples,tests]; extra == "all"
+Dynamic: license-file
 
 # Fake Convergence Beam Electron Diffraction (fakecbed)
 

fakecbed-0.3.4.dist-info/RECORD

@@ -0,0 +1,10 @@
+fakecbed/__init__.py,sha256=iP73WNV_HZgxiVUVdesbJqlrhDGESooh1hlBs8kox9M,1342
+fakecbed/discretized.py,sha256=8j2wFwGXNuf602cJui9ZsvbrGitnt6nAxDcNSbytekU,99658
+fakecbed/shapes.py,sha256=TiqJBZI4JZib0Q6zfO0e-dh7ACjYhvAYUvEtIedoo74,116471
+fakecbed/tds.py,sha256=mk5359ErIhdpCoTdHq9r1-foKb8t3qCOHqNoabx3bQw,8066
+fakecbed/version.py,sha256=Mxups7YfGBY2vvCok_hLJKCtU6O1WHQJfsMY-bAJ0Yg,511
+fakecbed-0.3.4.dist-info/licenses/LICENSE,sha256=N0P3pKtRMvfb64jmgJdlerg3TnS0a2QC776AzPsbZIg,35128
+fakecbed-0.3.4.dist-info/METADATA,sha256=EMbTje8PeaQkEs54LEIrruABi7a8QNrn73FjBDHLJ9M,3023
+fakecbed-0.3.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fakecbed-0.3.4.dist-info/top_level.txt,sha256=l0jL8P0CjRiCSznYLnq7riaAnOmYaV3zjjjIJx9vjSQ,9
+fakecbed-0.3.4.dist-info/RECORD,,

{fakecbed-0.3.2.dist-info → fakecbed-0.3.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

fakecbed-0.3.2.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=9jP8Fo8egXoMs_T3DFqSuJYg4n9o9mnwYubl_hnut4k,411
-fakecbed-0.3.2.dist-info/LICENSE,sha256=N0P3pKtRMvfb64jmgJdlerg3TnS0a2QC776AzPsbZIg,35128
-fakecbed-0.3.2.dist-info/METADATA,sha256=Jh0UTcfGaox7vLR_LUhIfdZeB7nzxrAR0gMK6rqFMcw,3006
-fakecbed-0.3.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-fakecbed-0.3.2.dist-info/top_level.txt,sha256=l0jL8P0CjRiCSznYLnq7riaAnOmYaV3zjjjIJx9vjSQ,9
-fakecbed-0.3.2.dist-info/RECORD,,