a better way of checking sizes

2026-07-17 11:32:59 +08:00 · 2015-05-07 15:29:41 -07:00
parent ce7b0b24da
commit 4bda4b6019
1 changed files with 2 additions and 2 deletions
@@ -160,8 +160,8 @@ class Fields(object):
                assert hasattr(self, func), 'The alias field function is a string, but it does not exist in the Fields class.'
                func = getattr(self, func)
            out = func(self._fields[alias][:,ind], srcII)
-            if not out.size > out.shape[0]:
-                out = Utils.mkvc(out)
+        if out.shape[0] == out.size:
+            out = Utils.mkvc(out)
        return out

    def __contains__(self, other):