DOC: Only define __getitem__ on NumPy 1.20+.

QuLogic · QuLogic · commit 3488dc7764c3 · 2021-03-25T04:18:01.000-04:00
On older versions (1.18), this breaks things, so must be skipped.
diff --git a/examples/units/basic_units.py b/examples/units/basic_units.py
@@ -5,6 +5,7 @@
 
 """
 
+from distutils.version import LooseVersion
 import math
 
 import numpy as np
@@ -154,8 +155,9 @@ def __str__(self):
     def __len__(self):
         return len(self.value)
 
-    def __getitem__(self, key):
-        return TaggedValue(self.value[key], self.unit)
+    if LooseVersion(np.__version__) >= '1.20':
+        def __getitem__(self, key):
+            return TaggedValue(self.value[key], self.unit)
 
     def __iter__(self):
         # Return a generator expression rather than use `yield`, so that
