HOTFIX: Numpy 1.19 compatibility

andykee · Aug 17, 2020 · d30fabf · d30fabf
1 parent 96e199e
commit d30fabf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -6,6 +6,7 @@ v0.3.3
 Released August 17, 2020
 
 * Make FPN seed optional in ``detector.dark_current``
+* Update ``np.ediff1d`` usage to be compatible with Numpy 1.19
 
 v0.3.2
 ------

diff --git a/lentil/detector.py b/lentil/detector.py
@@ -829,7 +829,7 @@ def _process_cube_intersections(inter, xyz, dxdydz):
     _dxdydz = dxdydz[inter['indx'], :].astype(np.float32)
 
     draylength = np.ediff1d(inter['raylength'], to_begin=(0.0,))
-    np.putmask(draylength, np.ediff1d(inter['indx'], to_begin=(1,)) != 0,
+    np.putmask(draylength, np.ediff1d(inter['indx'], to_begin=np.array([1], dtype=np.uint64)) != 0,
                inter['raylength'])
 
     inter_point = _xyz + inter['raylength'][:, None] * _dxdydz