Added dtype keyword to create_power_operator

40ed2574 · Theo Steininger · d658ea45 · 40ed2574
Commit 40ed2574 authored Mar 16, 2017 by Theo Steininger
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

nifty/sugar.py nifty/sugar.py +3 -1

No files found.
--- a/nifty/sugar.py
+++ b/nifty/sugar.py
@@ -8,12 +8,14 @@ from nifty import PowerSpace,\
 __all__ = ['create_power_operator']


-def create_power_operator(domain, power_spectrum, distribution_strategy='not'):
+def create_power_operator(domain, power_spectrum, dtype=None,
+                          distribution_strategy='not'):
    if not domain.harmonic:
        fft = FFTOperator(domain)
        domain = fft.target[0]

    power_domain = PowerSpace(domain,
+                              dtype=dtype,
                              distribution_strategy=distribution_strategy)

    fp = Field(power_domain,