series_expansion: catch IndexError as well

It seems numpy arrays throw this instead of TypeError.

1 files changed, 2 insertions, 2 deletions

diff --git a/series_expansion.py b/series_expansion.py
index 9cd7671..edee84e 100644
--- a/series_expansion.py
+++ b/series_expansion.py
@@ -19,7 +19,7 @@ class SeriesExpansion(object):
         if self._dim == 1:
             try:
                 basis[0]
-            except TypeError:
+            except (TypeError, IndexError):
                 basis = [basis]
 
         if len(basis) != coeffs.ndim:
@@ -41,7 +41,7 @@ class SeriesExpansion(object):
         if self._dim == 1:
             try:
                 coords[0][0]
-            except TypeError:
+            except (TypeError, IndexError):
                 coords = [coords]
 
         if diff_order is None: