diff options
author | Anton Khirnov <anton@khirnov.net> | 2014-10-22 14:57:55 +0200 |
---|---|---|
committer | Anton Khirnov <anton@khirnov.net> | 2014-10-22 14:57:55 +0200 |
commit | 89e1cc33f2a8a5c1d940c19ba7f6c71652948374 (patch) | |
tree | 305b433322c64526754811e0436ee5250711eb0d | |
parent | d50bf3e52dae076d07cdc139243266c8808654c6 (diff) |
series_expansion: allow passing scalar diff_order to 1-dim expansions
-rw-r--r-- | series_expansion.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/series_expansion.py b/series_expansion.py index edee84e..76adc30 100644 --- a/series_expansion.py +++ b/series_expansion.py @@ -44,6 +44,12 @@ class SeriesExpansion(object): except (TypeError, IndexError): coords = [coords] + if diff_order is not None: + try: + diff_order[0] + except (TypeError, IndexError): + diff_order = [diff_order] + if diff_order is None: diff_order = [0] * len(coords) |