Data.get_axes_from_description small fix

rwth-i6 · Dec 19, 2021 · 9120587 · 9120587
1 parent a929e28
commit 9120587
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/returnn/tf/util/data.py b/returnn/tf/util/data.py
@@ -4375,7 +4375,7 @@ def get_axes_from_description(self, axes, allow_int=NotSpecified):
       if len(dims) > 1:
         max_match_priority = max(self.dim_tags[i].match_priority for i in dims)
         dims = [i for i in dims if self.dim_tags[i].match_priority == max_match_priority]
-      assert len(dims) == 1, "%s: matching dim %s must be unique" % (self, axes)
+      assert len(dims) <= 1, "%s: matching dim %s must be unique" % (self, axes)
       return dims
     if isinstance(axes, int):
       self._verify_axis_int_from_description(allow_int=allow_int)