Fixed quadrature weight units in curvedsky.map2alm_cyl for partial sk…

…y coverage. Also added support for google collab notebook in enplot
simonsobs · Aug 3, 2023 · 4cc18a5 · 4cc18a5
1 parent 5b40a1f
commit 4cc18a5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pixell/curvedsky.py b/pixell/curvedsky.py
@@ -749,7 +749,7 @@ def map2alm_cyl(map, alm=None, ainfo=None, minfo=None, lmax=None, spin=[0,2], we
 			ny      = map.shape[-2]+np.sum(minfo.ypad)
 			weights = ducc0.sht.experimental.get_gridweights(minfo.ducc_geo.name, ny)
 			weights = weights[minfo.ypad[0]:len(weights)-minfo.ypad[1]]
-			weights/= map.shape[-1]
+			weights/= minfo.ducc_geo.nx
 		else:
 			weights = map.pixsizemap(separable=True, broadcastable=True)[:,0]
 		weights = weights.astype(map.dtype)

diff --git a/pixell/enplot.py b/pixell/enplot.py
@@ -952,7 +952,7 @@ def show(img, title=None, method="auto"):
 		# do. Try them one by one in priority order
 		try:
 			# Only use ipython for graphical notebooks
-			if "ZMQ" in get_ipython().__class__.__name__:
+			if get_ipython().__class__.__name__ in ["ZMQInteractiveShell", "Shell"]:
 				return show_ipython(img, title=title)
 		except (ImportError, NameError): pass
 		try: return show_tk(img, title=title)