From 770b34fc3d9de39b1ff207e64e0497c083780ead Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Tue, 8 Jun 2021 09:53:19 -0500
Subject: [PATCH] add arctan2
---
arraycontext/impl/pytato.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py
index d6d7bac..b6bf800 100644
--- a/arraycontext/impl/pytato.py
+++ b/arraycontext/impl/pytato.py
@@ -144,6 +144,10 @@ class _PytatoFakeNumpyNamespace(BaseFakeNumpyNamespace):
import pytato as pt
return rec_multimap_array_container(pt.conj, x)
+ def arctan2(self, y, x):
+ import pytato as pt
+ return rec_multimap_array_container(pt.arctan2, y, x)
+
# }}}
--
GitLab