Update af_backprop/alphafold/model/utils.py

af_backprop/alphafold/model/utils.py

@@ -53,7 +53,7 @@ def mask_mean(mask, value, axis=None, drop_mask_channel=False, eps=1e-10):
     axis = [axis]
   elif axis is None:
     axis = list(range(len(mask_shape)))
-  assert isinstance(axis, collections.Iterable), (
+  assert isinstance(axis, collections.abc.Iterable), (
       'axis needs to be either an iterable, integer or "None"')
 
   broadcast_factor = 1.